Member of Technical Staff, ML Infrastructure

Cognita Imaging • Palo Alto, CA

13h • Onsite

About The Position

As a Member of Technical Staff on the ML Infrastructure team, you will build and operate the platform that enables Cognita’s ML systems to scale reliably in training and production. Your focus will be on distributed systems, model serving, and the infrastructure that supports large-scale training and deployment of vision-language models used in clinical workflows.

Requirements

• Strong experience in ML infrastructure, MLOps, or distributed systems.
• Experience building or operating infrastructure for large-scale ML systems.
• Deep understanding of systems performance, reliability, and scalability.
• Ownership mindset and bias toward building robust long-lived platforms.
• Evidence of exceptional ability, demonstrated through previous builds.

Nice To Haves

• Experience with industry-standard distributed computing frameworks.
• Experience deploying large language or vision-language models.
• Experience with inference optimization techniques
• Experience with cloud infrastructure (e.g., AWS / GCP / Azure).
• Experience with GPU-accelerated infrastructure.
• Experience operating ML systems in production environments.

Responsibilities

• Build and operate distributed infrastructure for ML training and model serving at scale.
• Own systems that manage large-scale compute, storage, and data movement for ML workloads.
• Design and optimize model serving systems and APIs for extreme reliability, latency, and throughput.
• Support large distributed training runs by providing robust, scalable infrastructure primitives.
• Ensure infrastructure can handle large radiology studies, including multiple series and prior exams.
• Improve observability and reliability for ML systems in production.
• Work closely with ML training engineers to turn training requirements into scalable systems.
• Debug performance and reliability issues across the ML stack.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed