What are the responsibilities and job description for the Senior Research Engineer- Performance Optimization position at Luma AI?
We are looking for engineers with significant problem-solving experience in PyTorch, CUDA, and distributed systems. You will work with Research Scientists to build & train cutting-edge foundation models on thousands of GPUs.
Responsibilities
- Ensure efficient implementation of models & systems for data processing, training, inference, and deployment.
- Identify and implement optimization techniques for massively parallel and distributed systems.
- Identify and remedy efficiency bottlenecks (memory, speed, utilization) by profiling and implementing high-performance CUDA, Triton, C , and PyTorch code.
- Work closely together with the research team to ensure systems are planned to be as efficient as possible from start to finish.
- Build tools to visualize, evaluate, and filter datasets.
- Implement cutting-edge product prototypes based on multimodal generative AI.
Experience
175,000 - $250,000 a year
In addition to cash base pay, you'll also receive a sizable grant of Luma's equity.
The pay range for this position is for Bay Area. Base pay offered may vary depending on job-related knowledge, skills, candidate location, and experience.
Your applications are reviewed by real people.
J-18808-Ljbffr
Salary : $175,000 - $250,000
