Systems Research Engineer, GPU Programming
$160k - $230k • Remote • San Francisco
Posted 1mo ago
Job Location
San Francisco
Tech Stack
Remote Work Policy
Fully remote
Categories
Machine Learning Engineer
About the job
As a Systems Research Engineer specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. You will co-design GPU kernels and model architecture to enhance the performance and efficiency of our AI systems, and contribute to the co-design of efficient GPU architectures and programming models. Your research skills will be vital in staying up-to-date with the latest advancements in GPU programming techniques, ensuring that our AI infrastructure remains at the forefront of innovation.
Responsibilities
- Optimize and fine-tune GPU code for performance and scalability.
- Collaborate with cross-functional teams to integrate GPU-accelerated solutions.
- Stay updated on GPU programming techniques and technologies.
Requirements
- Strong background in GPU programming and parallel computing (e.g., CUDA, Triton).
- Knowledge of ML/AI applications and models.
- Familiarity with GPU performance profiling and optimization tools.
- Excellent problem-solving and analytical skills.
- Bachelor's, Master's, or Ph.D. in Computer Science, Electrical Engineering, or equivalent experience.
Benefits
- Competitive compensation
- Startup equity
- Health insurance
- Other benefits
- Flexibility in terms of remote work