Machine Learning Researcher

We're seeking a Machine Learning Researcher to push the boundaries of what's possible in LLM post-training. The successful candidate will conduct research into experimental models, training systems, and modalities to create novel products for our customers.

Requirements

• 3+ years of experience training AI models using PyTorch
• Deep understanding of transformer architectures, attention mechanisms, and model internals
• Hands-on experience with post-training LLMs using SFT, RLHF, DPO, or other alignment techniques
• Experience with LLM-specific training frameworks (e.g., Hugging Face Transformers, DeepSpeed, Megatron, TRL, or similar)
• Strong experimental methodology, including ability to design, run, and analyze rigorous experiments
• Track record of implementing ideas from recent ML papers
• Experience training on NVIDIA GPUs at scale
• Strong foundation in ML fundamentals: optimization, loss functions, regularization, generalization

Benefits

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