ML Systems Architect for Production-Grade AI

A1; Palo Alto (CA); On-site; USD 347,000 - 490,000; Full time

Job summary

A leading AI company in Palo Alto seeks a skilled individual to turn research into production-grade ML systems. The ideal candidate will build and maintain ML pipelines, fine-tune models, and collaborate with engineers to integrate these systems. This role requires a strong background in deep learning, hands-on experience with ML frameworks like PyTorch, and a bias toward rapid iteration. Compensation ranges from $347K to $490K, with an emphasis on technical expertise and team collaboration.

Qualifications

• Strong background in deep learning and transformer-based architectures.
• Hands-on experience training, fine-tuning, or deploying large-scale ML models in production.
• Proficiency with at least one modern ML framework, with a capacity to learn others quickly.

Responsibilities

• Build and own end-to-end ML pipelines from data to deployment.
• Architect and operate scalable inference systems balancing latency, cost, and reliability.
• Collaborate with engineering to integrate ML systems in products.

Skills

Deep learning fundamentals

Transformers-based architectures

Large-scale ML production

GPU optimization techniques

Iterative system improvements

Tools

PyTorch

JAX

DeepSpeed

Ray

Job description

A leading AI company in Palo Alto seeks a skilled individual to turn research into production-grade ML systems. The ideal candidate will build and maintain ML pipelines, fine-tune models, and collaborate with engineers to integrate these systems. This role requires a strong background in deep learning, hands-on experience with ML frameworks like PyTorch, and a bias toward rapid iteration. Compensation ranges from $347K to $490K, with an emphasis on technical expertise and team collaboration.