Machine Learning Research Engineer
RoboflowFull Time
Senior (5 to 8 years), Expert & Leadership (9+ years)
Key technologies and capabilities for this role
Common questions about this position
The salary range is $220K - $300K.
Yes, this is a remote position.
The role requires significant experience with PyTorch, strong experience with CUDA, experience with distributed systems, training large models using Python & PyTorch, profiling GPU & CPU code, writing highly parallel & distributed PyTorch code, and experience with transformer models.
You will work alongside the rest of the research team to build & train cutting edge foundation models and ensure systems are planned to be as efficient as possible.
Top candidates have experience with high-performance Triton/CUDA, writing custom PyTorch kernels, fused kernels for hot paths, high-performance parallel C++ in an ML context, and building inference/demo prototypes.
Develops multimodal AI technologies for creativity
Luma AI develops multimodal artificial intelligence technologies that enhance human creativity and capabilities. Their main product, the Dream Machine, allows users to interact with various types of data, enabling creative professionals, businesses, and developers to explore innovative applications of AI. Unlike many competitors, Luma AI focuses on integrating multiple modes of interaction, which broadens the possibilities for users. The company operates on a subscription model, providing access to its AI tools and services, and aims to lead the way in AI-driven creativity and productivity.