Senior ML Engineer (AI Research)
Core
Conducting applied AI research on reinforcement learning, guided search, and agent training for large-scale systems.
Role type
Senior IC machine-learning engineer (AI research)
Builds
Research findings applied to Nebius AI R&D and adjacent teams
Domain
AI research, reinforcement learning, large language models
Deliverable
research
Required skills
theoretical foundations of machine learning, reinforcement learning, deep learning for language processing, training large models on multiple nodes, Python, JAX, experimental design with statistical rigor, hypothesis formulation, technical documentation
Preferred skills
deep reinforcement learning for LLMs, LLM optimization techniques (RoPE, ZeRO/FSDP, Flash Attention, quantization), distributed data processing systems, open-source contributions
Technologies
Python, JAX
Responsibilities
Conduct experiments to train large language models on interaction traces, explore guided generation and search in trajectory space, mine web-scale data for model post-training, test reinforcement learning configurations in verifiable domains, train AI agents on tasks with non-verifiable reward signals
Seniority
Senior, hands-on IC