ML Infra Engineer: Scale Ray + PyTorch for Multimodal AI

Orbifold AI; Palo Alto (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

Orbifold AI is seeking a Machine Learning Engineer in Palo Alto, California, to build and optimize the infrastructure for multimodal data processing. You will play a key role in scaling our Ray and PyTorch-based systems utilized in frontier robotics.

The ideal candidate will have extensive experience with distributed systems, strong proficiency in Python, and proven knowledge of machine learning frameworks, particularly PyTorch. You will directly influence the efficiency and performance of our data processing and model training pipelines.

Qualifications

• 3+ years of software engineering experience with a focus on backend and distributed systems.
• Strong proficiency in production-grade Python code.
• Deep knowledge of PyTorch and experience with Ray.

Responsibilities

• Architect and optimize distributed ML pipelines on Ray and PyTorch for multimodal data.
• Profile and tune distributed training jobs for performance.
• Build internal tools for seamless PyTorch model deployment.

Skills

Backend development

Distributed systems

Machine Learning infrastructure

Python

PyTorch

Ray

Networking

Concurrency

Fault tolerance

Parallel processing

Tools

Kubernetes

Docker

AWS

GCP

Azure

FFmpeg

Job description

Orbifold AI is seeking a Machine Learning Engineer in Palo Alto, California, to build and optimize the infrastructure for multimodal data processing. You will play a key role in scaling our Ray and PyTorch-based systems utilized in frontier robotics.

The ideal candidate will have extensive experience with distributed systems, strong proficiency in Python, and proven knowledge of machine learning frameworks, particularly PyTorch. You will directly influence the efficiency and performance of our data processing and model training pipelines.