ML Solution Architect (Early Talent)

Summary

Nebius is leading a new era in cloud infrastructure for the global AI economy. They are seeking an ML Solutions Architect (Early Career) to assist in building and testing LLM-based solutions while collaborating closely with senior Solutions Architects.

Responsibilities

• Help build and test LLM-based solutions and applications using Token Factory's inference services, including multimodal models (text, vision, audio)
• Assist senior SAs with prompt engineering, model selection, benchmarking, and inference optimization
• Run performance and quality experiments to support proof-of-concept work
• Contribute to internal tooling and automation that improves how the SA team delivers

Skills

• Currently pursuing or recently completed a BSc/MSc/PhD in Computer Science, Machine Learning, or a related field
• Strong Python programming skills
• Hands-on generative AI experience, including with common ML frameworks (e.g., PyTorch, Transformers)
• Strong communication skills, with a willingness to explain technical concepts to diverse audiences
• Experience deploying/serving LLMs with vLLM, SGLang, or TensorRT-LLM
• Familiarity with inference optimization techniques such as quantization, batching, caching, and routing
• Knowledge of model architectures and fine-tuning approaches
• Contributions to open-source ML/AI projects

Qualifications

Must Haves

• Currently pursuing or recently completed a BSc/MSc/PhD in Computer Science, Machine Learning, or a related field
• Strong Python programming skills
• Hands-on generative AI experience, including with common ML frameworks (e.g., PyTorch, Transformers)
• Strong communication skills, with a willingness to explain technical concepts to diverse audiences

Nice to Haves

• Experience deploying/serving LLMs with vLLM, SGLang, or TensorRT-LLM
• Familiarity with inference optimization techniques such as quantization, batching, caching, and routing
• Knowledge of model architectures and fine-tuning approaches
• Contributions to open-source ML/AI projects

Benefits

• Health insurance: 100% company-paid medical, dental, and vision coverage for employees and families.
• 401(k) plan:Up to 4% company match with immediate vesting.
• Parental leave: 20 weeks paid for primary caregivers, 12 weeks for secondary caregivers.
• Remote work reimbursement: Up to $85/month for mobile and internet.
• Disability & life insurance: Company-paid short-term, long-term and life insurance coverage.