AI/ML Specialist Solutions Architect

Customer experience:

Customer experience at Nebius AI Cloud involves tackling customers’ challenges and directly impacting their success by solving real-world AI and ML problems at massive GPU cloud scale. You’ll not only resolve issues, but play a key role in shaping clients’ business success by optimizing their AI solutions. Working with advanced GPUs such as H200, B200 and GB200, as well as modern ML frameworks, you’ll influence the development of the Nebius AI Cloud and gain experience at the intersection of infrastructure and AI. With minimal bureaucracy, you’ll have the freedom to innovate, take ownership and drive change. Opportunities for growth are abundant in this vibrant and supportive professional community.

The role

We seek an experienced AI/ML Specialist Solutions Architect to support AI-focused customers leveraging Nebius services. In this role, you will be a trusted advisor, collaborating with clients to design scalable AI solutions, resolve technical challenges and manage large-scale AI deployments involving hundreds to thousands of GPUs.

You’re welcome to work remotely from the United States or Canada.

Your responsibilities will include:

• Designing customer-centric solutions that maximize business value and align with strategic goals.
• Building and maintaining long-term relationships to foster trust and ensure customer satisfaction.
• Delivering technical presentations, producing whitepapers, creating manuals and hosting webinars for audiences with varying technical expertise.
• Collaborating with engineering and product teams to effectively prioritize and relay customer feedback.

We expect you to have:

• Strong understanding of ML ecosystems, including models, use cases and tooling.
• Proven experience in setting up and optimizing distributed training pipelines across multi-node and multi-GPU environments.
• Hands-on knowledge of frameworks like PyTorch or JAX.
• Excellent verbal and written communication skills.

It will be an added bonus if you have:

• Expertise in deploying inference infrastructure for production workloads.
• Ability to transition ML pipelines from POC to scalable production systems.

Preferred tooling:

• Programming Languages – Python, Go, Java, C++
• Orchestration – Kubernetes (K8s), Slurm
• DevOps Tools – Git, Docker, Helm
• Infrastructure as Code (IaC) – Terraform
• ML Frameworks and Libraries – PyTorch, TensorFlow, JAX, HuggingFace, Scikit-learn

Key Employee 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.

Compensation

We offer competitive salaries, ranging from 215k - 275k OTE (On-Target Earnings) and equity based on your experience, skills, and location.

Join Nebius Today!