ML Solution Architect – Early Talent

Nebius Group

. Build and test LLM-based solutions and applications using Token Factory’s inference services, including multimodal text, vision, and audio models .

Posted 10/31/2026contractRemote • United StatesMid-LevelSenior💰 $102 - $126 per hourWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates strong Python programming skills and hands-on experience with generative AI, including proficiency in ML frameworks such as PyTorch and Transformers. Capable of collaborating with senior architects and contributing to the development of scalable AI applications while optimizing inference techniques.

Highest-signal resume keywords

Python ProgrammingGenerative AI ExperienceML Frameworks (PyTorch, Transformers)Inference Optimization TechniquesOpen-Source ML/AI Contributions

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

Python ProgrammingGenerative AIML FrameworksInference OptimizationModel ArchitecturesFine-Tuning ApproachesPerformance BenchmarkingPrompt EngineeringQuality ExperimentationAutomation Tools

Soft Skills

Strong Communication SkillsCollaborationMentorship

Tools & Technologies

VLLMSGLangTensorRT-LLMOpenAI/Anthropic SDKsLangChainLangSmithFastAPIFlaskKubernetesDocker

Industry Keywords

Computer ScienceMachine LearningAI ApplicationsInference ServicesMultimodal ModelsCloud Platforms (AWS, GCP, Azure)

Tech Stack

Tools & technologies

AWSAzureDockerFlaskGoogle Cloud PlatformKubernetesPythonPyTorch

About the role

Key responsibilities & impact

Build and test LLM-based solutions and applications using Token Factory’s inference services, including multimodal text, vision, and audio models
Assist senior Solutions Architects 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 Solutions Architect team delivery
Work alongside senior Solutions Architects with close mentorship
Collaborate closely with the backend team
Learn how scalable AI applications are built and tuned on Nebius’s platform

Requirements

What you’ll need

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 common ML frameworks such as PyTorch and Transformers
Strong communication skills, with willingness to explain technical concepts to diverse audiences
Permitted to work in the job’s location
Experience deploying/serving LLMs with vLLM, SGLang, or TensorRT-LLM (nice-to-have)
Familiarity with inference optimization techniques such as quantization, batching, caching, and routing (nice-to-have)
Knowledge of model architectures and fine-tuning approaches (nice-to-have)
Contributions to open-source ML/AI projects (nice-to-have)
Preferred technical stack includes Python; vLLM, SGLang, TensorRT-LLM, Transformers, OpenAI/Anthropic SDKs; LangChain, LangSmith, smolagents or equivalent; FastAPI; Flask; Kubernetes; Docker; Git; AWS, GCP, and Azure AI platforms

Benefits

Comp & perks

100% company-paid medical, dental, and vision coverage for employees and families
401(k) plan with up to 4% company match and immediate vesting
20 weeks paid parental leave for primary caregivers
12 weeks paid parental leave for secondary caregivers
Remote work reimbursement up to $85/month for mobile and internet
Company-paid short-term, long-term, and life insurance coverage
Competitive compensation
Career growth and learning opportunities
Flexibility and ownership
Collaborative and innovative culture
Opportunity to work on impactful AI projects
International environment and talented teams