Remote ML Solutions Architect — Early Career

Nebius Group; United States; Remote; USD 102,000 - 126,000; Part time

Benefits offered by this job

Health insurance

401(k) plan

Parental leave

Remote work reimbursement

Disability & life insurance

Job summary

Nebius Group is seeking an ML Solutions Architect (Early Career) for a paid, 3-month contract, remote from the USA. You will work with senior Solutions Architects on building and testing LLM-based solutions, benchmarking, and inference optimization, gaining hands-on experience in a fast-paced AI platform.

The role emphasizes mentorship and real project work with opportunities to transition to a full-time position for top performers.

Qualifications

• 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.

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

Python programming

Machine Learning

LLMs / Open-source LLMs

Prompt engineering

Communication skills

Education

BSc/MSc/PhD in Computer Science or related

Tools

PyTorch

Transformers

vLLM

SGLang

TensorRT-LLM

Job description

Nebius Group is seeking an ML Solutions Architect (Early Career) for a paid, 3-month contract, remote from the USA. You will work with senior Solutions Architects on building and testing LLM-based solutions, benchmarking, and inference optimization, gaining hands-on experience in a fast-paced AI platform.

The role emphasizes mentorship and real project work with opportunities to transition to a full-time position for top performers.