Senior ML Engineer (AI Research)

We expect you to have

• A profound understanding of theoretical foundations of machine learning and reinforcement learning
• Deep expertise in modern deep learning for language processing and generation
• Substantial experience with training large models on multiple computational nodes
• Strong software engineering skills (we mostly use python)
• Deep experience with modern deep learning frameworks (we use jax)
• Strong communication and leadership abilities
• Ability to formulate research questions, design experiments to test hypotheses, and draw meaningful conclusions from results
• Ability to document research findings clearly and contribute to technical publications or report
• Experience with deep reinforcement learning for LLMs, including techniques such as reward modeling, DPO, PPO etc
• Familiarity with important ideas in LLM space, such as RoPE, ZeRO/FSDP, Flash Attention, quantization
• Bachelor’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field Master’s or PhD preferred
• Track record of building and delivering products (not necessarily ML-related) in a dynamic startup-like environment

Benefits & Perks

• Competitive compensation
• Career growth and learning opportunities
• Flexibility and ownership
• Collaborative and innovative culture
• Opportunity to work on impactful AI projects
• International environment and talented teams

What's it like to work at Nebius