about 2 hours ago
Base Salary
$295k - $345k/yr
Responsibilities
- Optimize machine learning models for performance on GPU architectures.
- Conduct low-level performance profiling analysis to identify bottlenecks.
- Contribute to the development of best practices and tooling for model optimization.
- Collaborate with cross-functional teams to integrate and deploy optimized models.
- Build tooling, interfaces, and visualizations for ML@Roblox.
Requirements
- 6+ years of professional experience in system design.
- Significant experience debugging GPUs and reading GPU profiles.
- Proficient in advanced tools and frameworks like CUDA, Triton, and TensorRT.
- Experience with model optimization techniques for LLMs.
- Passionate about enhancing model performance and building supportive tools.
- Bachelor's degree in Computer Science, Computer Engineering, Data Science, or a similar field.
Categories
AI & MLData Science