Senior ML Solutions Architect - Token Factory
Why work at Nebius
Nebius is leading a new era in cloud computing to serve the global AI economy. We create the tools and resources our customers need to solve real-world challenges and transform industries, without massive infrastructure costs or the need to build large in-house AI/ML teams. Our employees work at the cutting edge of AI cloud infrastructure alongside some of the most experienced and innovative leaders and engineers in the field.
Where we work
Headquartered in Amsterdam and listed on Nasdaq, Nebius has a global footprint with R&D hubs across Europe, North America, and Israel. The team of over 800 employees includes more than 400 highly skilled engineers with deep expertise across hardware and software engineering, as well as an in-house AI R&D team.
The role
We seek an experienced Senior ML Solutions Architect to support customers leveraging Nebius Token Factory's serverless inference platform for open-source LLMs across multiple modalities. In this role, you will be collaborating with clients to design and implement customized LLM-based solution and architect scalable AI applications using our served models, and working together with our backend team to improve our platform to match the clients' needs.
Your responsibilities will include:
- Design and implement LLM-based solutions using Nebius Token Factory’s inference services to drive business value and support customer goals.
- Build production-ready applications leveraging our serverless LLM APIs, including multimodal models (text, vision, audio) and domain-specific models.
- Provide technical expertise in prompt engineering, RAG architectures, model selection, and inference optimization.
- Collaborate with product and engineering teams to surface customer feedback and shape the platform roadmap.
- Guide customers in scaling from POC to production with a focus on performance, reliability, and cost efficiency.
We expect you to have:
- 5+ years of experience in ML/AI systems, with at least 2 years focused on LLMs and generative AI.
- Deep knowledge of the LLM ecosystem, including model architectures and fine-tuning approaches.
- Hands-on experience with:
- Prompt engineering and LLM pipeline development, including evaluation.
- Agentic frameworks such as Langchain, Langsmith, smolagents, or equivalent.
- Vector databases and RAG implementation patterns.
- Deploying LLM-powered applications using APIs from OpenAI, Anthropic, or open-source models.
- Strong Python programming skills.
- Excellent communication skills, with the ability to clearly explain technical concepts to diverse audiences.
It will be an added bonus if you have:
- Experience with inference frameworks and libraries (e.g., vLLM, SGLang, TensorRT-LLM, Transformers).
- Familiarity with inference optimization techniques such as quantization, batching, caching, and routing.
- Work with multimodal AI models (e.g., vision-language, speech).
- Proficiency with DevOps tools (Docker, Kubernetes).
- Contributions to open-source ML/AI projects.
Preferred tooling:
- Programming Languages– Python
- ML Frameworks and Libraries– vLLM, SGLang, TensorRT-LLM, Transformers, OpenAI/Anthropic SDKs
- Frameworks for Agentic Pipelines : Langchain / Langsmith / smolagents / equivalent
- API and Web Frameworks– FastAPI, Flask
- MLOps and DevOps tools– Kubernetes (K8s), Docker, Git
- Cloud Platforms– AWS (SageMaker, Bedrock), GCP (Vertex AI), Azure (Azure ML)
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!
What we offer
- 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.
We’re growing and expanding our products every day. If you’re up to the challenge and are excited about AI and ML as much as we are, join us!
Create a Job Alert
Interested in building your career at Nebius? Get future opportunities sent straight to your email.
Apply for this job
*
indicates a required field