Lead machine learning engineer (Model Architectures & Capabilities Team)
Nebius
- London
- Permanent
- Full-time
- Lead the team responsible for model architectures and capabilities
- Define strategy and tactics, i.e. figure out what research and engineering directions to pursue to push the technology forwards, and help the team plan and execute the experiments that will get us there efficiently
- Ensure high standards of engineering and research activities within the team
- Keep improving the design our internal infrastructure for training large models to ensure it keeps being fast and flexible despite the technology moving forwards
- Mentor our engineers and researchers
- A profound understanding of theoretical foundations of machine learning
- Deep expertise in modern deep learning for language processing and generation
- Substantial experience with pre-training large models on huge clusters
- Good understanding of performance aspects of large neural network training (sharding strategies, custom kernels, hardware features etc.)
- Strong software engineering skills (we mostly use python)
- Deep experience with modern deep learning frameworks (we use jax)
- Proficiency in contemporary software engineering approaches, including CI/CD, version control, and unit testing
- Strong communication and leadership abilities
- 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
- Experience in engineering complex systems, such as large distributed data processing systems or high-load web services
- Open-source projects that showcase your engineering prowess
- Excellent command of the English language, alongside superior writing, articulation, and communication skills