Applied Machine Learning Engineer

Help us build the systems that train specialized AI models for the fastest-growing companies in the world. We're a well-funded ten-person team of engineers who work in-person in downtown San Francisco on difficult, high-impact engineering problems.

Requirements

• 2+ years of experience training AI models using PyTorch
• Hands-on experience with post-training LLMs using SFT or RL
• Strong understanding of transformer architectures and how they're trained
• Experience with LLM-specific training frameworks
• Experience training on NVIDIA GPUs
• Strong data processing skills and comfortable building ETL pipelines and working with large datasets
• Track record of creating benchmarks and evaluations
• Ability to take research techniques and apply them to production systems

Benefits

• Competitive compensation
• Equity in a high-growth startup
• Comprehensive benefits