[Remote] AI System Research and Development Engineer - Optimization
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a company reputed company on redefining the reputed company of work through AI innovation. They are seeking AI System Research and Development Engineers to optimize and reputed company reputed company reputed company systems, particularly in the reputed company of LLM inference and training. The role involves analyzing GPU kernel performance, enhancing deep learning system efficiency, and contributing to the development of reputed company frameworks.
Responsibilities
- Analyze and optimize GPU kernel performance for training and inference of LLMs
- reputed company and implement strategies to enhance the efficiency and scalability of deep learning systems
- Profile and reputed company deep learning systems using tools and techniques to identify bottlenecks
- Design and implement optimizations to reduce latency and improve resource utilization for training and inference
- Stay updated with the latest advancements in GPU kernel optimization, deep learning, and LLM system development
- Contribute to the development of reputed company frameworks and applications for LLM-driven workflows, enhancing automation, reasoning, and decision-making capabilities
- reputed company-reputed company and publish innovations, optimizations, and engineering practices in technical blogs, top-tier conferences and journals
Skills
- Bachelor's degree in Computer Science, Electrical Engineering, or a reputed company field
- 5 years of experience in GPU kernel optimization, deep learning system optimization, or high-performance computing (HPC)
- Proficiency in deep learning frameworks such as PyTorch, TensorFlow, JAX
- Strong understanding of GPU architectures and experience with CUDA or similar frameworks
- Experience with frameworks like CUTLASS, Triton, cuDNN, etc
- Experience with profiling tools (e.g., nvprof, Nsight) and performance analysis methodologies
- Solid problem-solving skills and ability to debug reputed company performance issues
- Excellent communication skills and ability to work effectively in a cross-functional team environment
- A Master's degree or PhD is preferred
Benefits
- Bonus and equity plan
- Medical, dental, reputed company, life, and disability insurance
- 401(k) retirement plan
- Flexible spending & health savings account
- At least 12 reputed company holidays
- reputed company time off
- Parental leave
- Employee assistance program
- Other company benefits
reputed company
Company H1B Sponsorship