Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home - G

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

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home

As Machine Learning Technical Lead, you own the execution layer of intelligence. You will translate research direction into reliable, scalable, production-grade ML systems. This role sits at the intersection of research, infrastructure, and product. You will be responsible for making models trainable, deployable, observable, and performant under real-world constraints.

This position is 100% Remote.MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT.

Machine Learning Technical Lead Responsibilities:

• Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, and deployment.

• Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation.

• Architect and operate scalable inference systems, balancing latency, cost, and reliability.

• Design and maintain data systems for high-quality synthetic and real-world training data.

• Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership.

• Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.

• Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products.

• Make pragmatic trade-offs and ship improvements quickly, learning from real usage.

• Work under real production constraints: latency, cost, reliability, and safety

Machine Learning Technical Lead Outcomes:

• Research and models reliably translate into production-ready solutions with clear performance and quality targets.

• ML pipelines, training loops, and inference systems are stable, efficient, and maintainable.

• Production issues are detected, debugged, and resolved quickly, minimizing user impact.

• Team members are supported, aligned, and able to deliver high-impact ML work with minimal friction.

• Iterations on models and systems are measurable, safe, and improve user experience over time.

Machine Learning Technical Lead Qualifications:

• Experience building or shipping real Machine Learning systems used by people, not just demos.

• Artificial Intelligence (AI) experience required.

• Experience working with large models and understanding their failure modes.

• Experience writing strong, production-grade code.

• You are self-directed, pragmatic, and take full ownership of outcomes.

• You communicate clearly and collaborate well in small, high-trust teams.

• Tech Stack: GPU-based training and inference system, JAX, Python, and PyTorch.

Benefits include medical insurance, Dental, Vision, Savings Plan Options, PTO, etc.

Keywords: San Francisco CA Jobs, Machine Learning Technical Lead, AI, Artificial Intelligence, Distillation, DPO, GPU Optimization, JAX, LoRA, Machine Learning, Python, PyTorch, QLoRA, SFT, Technical Lead, Work From Home, Remote, California Recruiters, IT Jobs, California Recruiting