Remote ML Systems Lead - Production-Grade AI Architect

Next Step Systems; United States; Remote; USD 150,000 - 230,000; Full time

Benefits offered by this job

Medical insurance

Dental

Vision

Savings plan

Job summary

Next Step Systems is seeking a Machine Learning Technical Lead to own the execution layer of intelligence. This 100% remote role translates research directions into scalable ML systems, bridging research, infrastructure, and product, with ownership of end-to-end pipelines.

You will train, deploy, observe, and optimize models on GPU-based systems (JAX, PyTorch, Python), collaborate with backend teams, and ship robust, production-grade code under real constraints.

Qualifications

• Experience building or shipping real ML systems used by people.
• AI experience required.
• Experience with large models and failure modes.
• Proficient in production-grade code.
• Self-directed, pragmatic, takes ownership of outcomes.
• Clear communication and collaboration in small teams.
• Tech stack includes GPU-based training/inference with JAX, Python and PyTorch.

Responsibilities

• Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, deployment.
• Fine-tune and adapt models using state-of-the-art methods (LoRA, QLoRA, SFT, DPO, distillation).
• Architect scalable inference systems balancing latency, cost, reliability.
• Design and maintain data systems for high-quality training data.
• Implement evaluation pipelines covering performance, robustness, safety, bias.
• Own production deployment, including GPU optimization and latency reduction.
• Collaborate with application engineering to integrate ML into backend/mobile/desktop products.
• Make pragmatic trade-offs and ship improvements quickly.
• Work under real production constraints: latency, cost, reliability, safety.

Skills

Real ML systems

AI experience

Large models

Production-grade code

Ownership

Clear communication

GPU training

Tools

JAX

PyTorch

Python

GPU hardware

Job description

Next Step Systems is seeking a Machine Learning Technical Lead to own the execution layer of intelligence. This 100% remote role translates research directions into scalable ML systems, bridging research, infrastructure, and product, with ownership of end-to-end pipelines.

You will train, deploy, observe, and optimize models on GPU-based systems (JAX, PyTorch, Python), collaborate with backend teams, and ship robust, production-grade code under real constraints.