Staff ML Engineer — Proactive AI Systems (Remote) + Equity

USA Tech Recruit; San Francisco (CA); On-site; USD 180,000 - 230,000; Full time

Job summary

USA Tech Recruit is assisting a startup in San Francisco to hire a Staff Machine Learning Engineer for a remote-friendly role. You will build core ML systems powering a proactive AI assistant designed to complete multi-step tasks with minimal prompting.

Equity is included. You will own data prep, training, evaluation, inference, and iteration; ship and measure outcomes; collaborate with research, product, and engineering; mentor peers and operate under production constraints like latency and

Qualifications

• Experience shipping ML systems used by real users.
• Understanding of how ML models behave in production.
• Ability to write robust, scalable code.

Responsibilities

• Build core ML systems that power a proactive AI product.
• Own end-to-end work: data prep, training, evaluation, inference, iteration.
• Turn research ideas into working production systems.
• Debug model failures and system issues using production signals.
• Iterate quickly: ship, measure outcomes, refine, and repeat.
• Collaborate with research, product, and engineering to deliver user impact.
• Mentor and review work from other ML engineers.
• Work under production constraints: latency, cost, reliability, and safety.

Skills

Python

Production-quality code

Systems thinking

Independent ownership

Communication

Tools

PyTorch

JAX

GPU-based training

Job description

USA Tech Recruit is assisting a startup in San Francisco to hire a Staff Machine Learning Engineer for a remote-friendly role. You will build core ML systems powering a proactive AI assistant designed to complete multi-step tasks with minimal prompting.

Equity is included. You will own data prep, training, evaluation, inference, and iteration; ship and measure outcomes; collaborate with research, product, and engineering; mentor peers and operate under production constraints like latency and