Senior ML Engineer (AI Research)
Core
Senior ML Engineer focused on applied AI research, developing advanced intelligent systems including reinforcement learning, agentic systems, and reasoning models.
Role type
Senior IC machine learning engineer (AI research)
Builds
Advanced AI capabilities, large-scale models, and research prototypes
Domain
Artificial Intelligence, Deep Learning, Reinforcement Learning, Large Language Models
Deliverable
production ML models
Required skills
Machine learning engineering, applied AI research, reinforcement learning, deep learning for language processing, large-scale distributed training, Python, JAX, experimental design, statistical rigor, technical documentation
Preferred skills
Deep reinforcement learning for LLMs (reward modeling, DPO, PPO), modern LLM techniques (RoPE, ZeRO/FSDP, Flash Attention, quantization), startup environment experience, large-scale distributed data processing
Technologies
Python, JAX, distributed training frameworks
Responsibilities
Conduct applied ML research in guided search, RL, agentic systems, and model distillation; Design and execute experiments for training LLMs using interaction traces; Explore methods for guided generation and search in model trajectory spaces; Investigate web-scale data collection and mining; Develop efficient post-training workflows; Experiment with RL configurations in verifiable and non-verifiable reward domains; Formulate research questions and translate hypotheses into experiments; Analyze results with statistical rigor; Develop and train large-scale models across multiple nodes; Implement research using modern deep learning frameworks; Translate findings into practical solutions; Contribute to technical publications and documentation; Shape research directions and mentor teams; Collaborate on scalable experimentation systems; Contribute to engineering practices for reproducible research
Seniority
Senior, hands-on IC with technical leadership