PnP Optimization Engineer
WaabiFull Time
Senior (5 to 8 years), Expert & Leadership (9+ years)
Key technologies and capabilities for this role
Common questions about this position
The pay range for this position in California is $180,000 - $250,000 per year.
This is a hybrid position.
Required skills include significant experience in PyTorch, CUDA, and distributed systems; proficiency in writing custom PyTorch kernels and high-performance C++; and experience with the full ML pipeline including data processing, training, and inference optimization.
The role includes competitive equity packages in the form of stock options and a comprehensive benefits plan.
Strong candidates have hands-on experience training large models with PyTorch, optimizing distributed systems using DDP/FSDP/Tensor Parallel, and writing high-performance CUDA/C++ code, with bonus for deep learning knowledge in Transformers and multimodal models.
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.