Senior ML Engineer – Token Factory
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Senior ML Engineer at Nebius developing an inference platform for fast, reliable training & deployment of LLMs. Collaborating in a dynamic AI cloud infrastructure environment.
Responsibilities:
- Enhance fine-tuning methodologies - both LoRA-based and full-parameter - for cutting-edge LLMs (e.g., GPT-OSS, Kimi K2, DeepSeek V3/R1, GLM-4.5), focusing on both model quality and training efficiency.
- Research and implement advanced inference optimization techniques, such as speculative decoding, quantization, and large-scale draft model training.
- Re-implement state-of-the-art open-source LLM architectures in JAX.
Requirements:
- 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.
- Reasonable 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.
- Previous experience working with language models or other similar NLP technologies is nice to have.
- Familiarity with important ideas in LLM space, such as MHA, RoPE, ZeRO/FSDP, Flash Attention, quantization is a plus.
- A track record of building and delivering products (not necessarily ML-related) in a dynamic startup-like environment is desired.
- Strong engineering skills, including experience in developing large distributed systems or high-load web services is beneficial.
- Open-source projects that showcase your engineering prowess would be an advantage.
- Excellent command of the English language, alongside superior writing, articulation, and communication skills is required.
Benefits:
- Competitive salary and comprehensive benefits package.
- Opportunities for professional growth within Nebius.
- Flexible working arrangements.
- A dynamic and collaborative work environment that values initiative and innovation.
Similar Jobs
Senior Manager – Machine Learning
Senior Manager leading the Machine Learning team to enhance CCC’s subrogation products. Collaborating with stakeholders and delivering innovative AI solutions in the insurance technology space.
Senior AI/ML Engineer
Senior AI/ML Engineer designing and implementing workflows leveraging LLMs at iBusiness. Collaborating with cross-functional teams to develop scalable AI-driven solutions.
Machine Learning Engineer – Mid-level
Machine Learning Engineer responsible for developing and operating scalable ML solutions at Extractta. Collaborating with product, data, and engineering teams to deliver robust models.
Senior Machine Learning Engineer
Senior Machine Learning Engineer at IQVIA leading ML applications design focusing on generative AI solutions. Collaborating with teams and guiding development of production-grade algorithms.
Senior Machine Learning Engineer
Senior Machine Learning Engineer developing core ML models and data pipelines for a gaming company. Seeking an autonomous engineer with deep passion for video games.
Machine Learning Engineer
Machine Learning Engineer developing AI safety solutions at Gray Swan. Focus on real-world impact in adversarial testing and model evaluation.
Staff Machine Learning Engineer – Generative AI, Full-Stack Applications
Senior Machine Learning Engineer developing AI solutions and full-stack applications for CVS Health. Collaborating with teams to design and prototype secure, resilient products and platform components.
Senior Machine Learning Engineer – Consultant
Senior AI/ML Engineer Consultant responsible for implementing AI solutions and mentoring juniors at leading AI consultancy Provectus. Join to solve complex technical challenges and impact enterprise clients.
Machine Learning Engineer
Machine Learning Engineer designing and evaluating deep learning models for agricultural robotics applications. Collaborating with engineers and product teams to enhance model performance in real-world scenarios.
Senior Machine Learning Engineer – AI Product
Machine Learning Engineer for AI Product team at Qonto, developing customer-facing AI solutions. Collaborating with stakeholders and ensuring integration of ML solutions into financial services.
ML Engineer – Agents, Data
Senior ML Engineer for AI product development in a leading social discovery company. Working on large-scale ML systems and cutting-edge generative AI technologies.
Senior MLOps
MLOps role enhancing ML infrastructure and deployment for a global travel-tech company. Focus on managing ML services and optimizing resources in a fully remote setup.
MLOps Engineer
ML Ops Engineer designing and maintaining ML workflows using AWS services. Collaborating with data scientists to ensure reliable and scalable models in production.
Machine Learning Lead
Machine Learning Lead driving AI initiatives for LexisNexis Legal & Professional, delivering scalable, robust solutions with hands-on leadership and mentoring.
Senior ML Engineer
Senior AI Engineer contributing to AI engineering teams at Relativity. Design and develop scalable AI systems, collaborating with cross-functional teams and driving engineering excellence.
AI/ML Engineer
AI/ML Engineer developing AI solutions for professional services firms. Working on production models and real data to enhance client relationships.
Machine Learning Engineer
Machine Learning Engineer developing AI capabilities within Zillow’s Agentic AI team. Focusing on production-grade systems that enhance real estate experiences through advanced reasoning.
Senior Machine Learning Engineer – ML/AI Team
Senior Machine Learning Engineer at A Place for Mom focused on developing ML and AI solutions. Designing and implementing advanced AI models to drive business impact using unstructured data.
Senior Machine Learning Engineer
Senior Machine Learning Engineer responsible for the full ML lifecycle at Button. Collaborating with data teams to build systems that integrate machine learning into commerce and monetization products.
Senior Machine Learning Engineer
Senior Machine Learning Engineer developing AI solutions for customer service at Replicant. Leading exploration of Large Language Models and Generative AI for innovative user experiences.