Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Member of Technical Staff, ML Infrastructure

Cognita Imaging Inc.

About Cognita Cognita’s mission is to increase the world’s access to healthcare. Radiology is (1) the first-line diagnostic specialty, (2) facing a worsening global workforce shortage, and (3) highly digitized, making it uniquely positioned for AI to have an enormous impact. Stage one of Cognita is focused on expanding access to radiology at scale. Our founding team met at Stanford, where they laid the groundwork for applying comprehensive AI to radiology. Building on that foundation, Cognita develops vision-language models that read radiology studies the way radiologists do – interpreting the full study in clinical context – and generate draft results that make radiologists more efficient and accurate. In partnership with Radiology Partners, Cognita’s models are trained and validated on one of the world’s largest real-world radiology datasets. About the Role 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. Your Impact 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 You Bring We are not credential-driven. We only look for evidence of exceptional ability. Please show us what you have built previously. 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. What Sets You Apart Experience with industry-standard distributed computing frameworks. Experience deploying large language or vision-language models. Experience with inference optimization techniques Experience with cloud infrastructure (AWS/GCP/Azure). Experience with GPU-accelerated infrastructure. Experience operating ML systems in production environments. Our Culture We’re an in-person team based in the San Francisco Bay Area. Working together every day helps us move faster, learn from each other, and build strong relationships. We believe the best work happens when great people are in the same room. We build from first principles. We question assumptions, reason from fundamentals, and execute with speed and clarity, without sacrificing quality. We operate as a true meritocracy. Impact matters, rather than optics. Feedback is direct, respect is non-negotiable, and we support each other with the ownership, trust, and resources needed to do the best work of our careers. Every decision we make is grounded in supporting clinicians to improve patient outcomes. #J-18808-Ljbffr Cognita Imaging Inc.

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Member of Technical Staff, ML Infrastructure in Palo Alto, CA vacancy
  • Member of Technical Staff - Foundation Model Architecture & AI Infrastructure Vinci | Full-Time | Remote / Hybrid The Mission At Vinci, we are building the operator intelligence...  ...) Distributed training systems Production ML system design Scaling structured datasets... 
    Suggested
    Full time
    Remote work

    Vinci4d

    Palo Alto, CA
    5 days ago
  •  ...layer for the hardware revolution. What You'll Do As a Founding Member of the Technical Staff (ML infra) at Architect, you'll be responsible for the critical algorithms and infrastructure that our researchers depend on to train models. Your work will directly enable... 
    Suggested

    Architect Labs

    Palo Alto, CA
    1 day ago
  • $180k

    Member Of Technical Staff - Cloud Infrastructure ABOUT xAI xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization... 
    Suggested
    Temporary work

    x.ai

    Palo Alto, CA
    2 days ago
  • $180k

     ...at X builds and operates the infrastructure responsible for all large-scale...  ...Trino, enabling real-time ML pipelines, feed ranking, experimentation...  ...interview”) during which a member of our team will ask some...  ...process, which consists of 2 technical interviews and 1 project deep... 
    Suggested
    Temporary work
    H1b
    Work at office
    Work from home
    Work visa

    xAI

    Palo Alto, CA
    1 day ago
  • $180k - $250k

    Member of Technical Staff -- TPU Systems (JAX / XLA / PALLAS) About the Role RadixArk is looking for...  ...working on SGLang-JAX and other critical infrastructure that enables efficient deployment of...  ...years experience building production ML systems with JAX, XLA, or TPU-focused... 
    Suggested
    Full time
    Flexible hours

    RadixArk

    Palo Alto, CA
    1 day ago
  • $180k

    Member of Technical Staff - Multimodal Understanding About xAI xAI’s mission is to create AI systems...  ...pre‑training, post‑training/alignment, infrastructure/scaling, evaluation, tooling/demos,...  ...or optimizing large‑scale distributed ML systems (training/inference... 
    Temporary work

    xAI

    Palo Alto, CA
    2 days ago
  • $180k

     ...at X builds and operates the infrastructure responsible for all large‑scale...  ...Trino, enabling real‑time ML pipelines, feed ranking, experimentation...  ...interview”) during which a member of our team will ask some...  ...process, which consists of 2 technical interviews and 1 project deep... 
    Temporary work
    Work at office
    Work from home

    Pantera Capital

    Palo Alto, CA
    1 day ago
  • Member of Technical Staff -- Cluster / Platform About the Role RadixArk is looking for a Member of Technical...  ...limits of large-scale distributed infrastructure for AI workloads. This role focuses...  ...Plus: Experience with large-scale ML/AI workloads Familiarity with RDMA,... 
    Flexible hours

    RadixArk

    Palo Alto, CA
    1 day ago
  • RadixArk is seeking a Member of Technical Staff — Training to build and scale the systems that train...  ...on large-scale distributed training infrastructure for LLMs and generative models, pushing...  ...role sits at the intersection of ML, systems, and performance engineering... 
    Flexible hours

    RadixArk

    Palo Alto, CA
    3 days ago
  • Member of Technical Staff — Supercomputing About the Role RadixArk is hiring a Member of Technical Staff...  ..., and operate production‑grade AI infrastructure for frontier‑scale inference and...  ...infrastructure, LLM inference serving, ML systems, or large‑scale training workloads... 
    Flexible hours

    RadixArk

    Palo Alto, CA
    3 days ago
  • $180k

     ...software engineers to build the software infrastructure that enables our models to use tools...  ...interview (“phone interview”) during which a member of our team will ask some basic...  ...the main process, which consists of four technical interviews: Coding assessment in a language... 
    Local area
    Relocation

    Pantera Capital

    Palo Alto, CA
    5 days ago
  •  ...of AI silicon, systems, software, and infrastructure to build the next generation of full-...  .... We are looking for an exceptional Member of Technical Staff to help design, build, and scale core...  ..., distributed systems, compilers, ML systems, hardware‑software co‑design,... 

    DensityAI

    Mountain View, CA
    1 day ago
  • Member of Technical Staff (Data Acquisition) About the Role Your mission is to build and operate the...  ...operate large-scale distributed crawling infrastructure capable of continuously discovering...  ...data processing, architecture, and ML platform teams to ensure smooth data... 

    Sanas

    Palo Alto, CA
    5 days ago
  •  ...real-world radiology datasets. About the Role As a Member of Technical Staff focused on ML Evaluation, you will be responsible for understanding...  ...and generative models. Partner with ML training and infrastructure teams to guide model improvement. Work with clinicians... 

    Cognita Imaging Inc.

    Palo Alto, CA
    1 day ago
  •  ...Li (Godmother of AI). We're building the infrastructure for a new era of interactive...  ...lasts two hours. As our RecSys founding member, you'll own this problem end-to-end - set...  ...of experience in recommendation systems, ML engineering, or applied ML in a consumer... 

    Astrocade

    Palo Alto, CA
    3 days ago
  • RadixArk is seeking a Member of Technical Staff — Inference to push the limits of large-scale AI inference. You will work on the core systems...  ...role sits at the intersection of systems engineering, ML infrastructure, and performance optimization. Your work will directly... 
    Worldwide
    Flexible hours

    RadixArk

    Palo Alto, CA
    1 day ago
  • About the Role As a Member of Technical Staff [Research] at NeoCognition , you’ll be part of the core...  ...Python and familiarity with modern ML frameworks (e.g., PyTorch, JAX, or TensorFlow...  ..., Mistral, or similar) and training infrastructure. Publications in top-tier AI venues... 

    NeoCognition Inc.

    Palo Alto, CA
    1 day ago
  • Member of Technical Staff — Kernel / Compiler / Communication About the Role RadixArk is seeking a Member...  ...Contributions to kernel/compiler/ML systems open source Experience...  ...design About RadixArk RadixArk is an infrastructure‑first AI company built by engineers who... 
    Flexible hours

    RadixArk

    Palo Alto, CA
    4 days ago
  •  ...Role We are seeking a highly skilled Member of Technical Staff to join our team in managing and enhancing...  ...operations for mission-critical AI infrastructure. The ideal candidate will combine...  ...ideally in hyperscale, cloud, or AI/ML training infrastructure environments... 

    x.ai

    Palo Alto, CA
    3 days ago
  •  ...largest real-world radiology datasets. About the Role As a Member of Technical Staff focused on model training, you will be responsible for...  ...regressions, and failure modes. Work closely with ML infrastructure engineers to ensure training systems are reliable, performant... 

    Cognita Imaging Inc.

    Palo Alto, CA
    1 day ago
  • Member of Technical Staff Physical AI (Robotics / World Models) Palo Alto, CA About Orbifold AI Orbifold...  ...Closing that gap requires evaluation infrastructure that can systematically surface,...  ...experience building or iterating on applied ML systems at scale Comfortable... 
    Shift work

    Bonfirevc

    Palo Alto, CA
    4 days ago
  • $90k - $130k

    Member of Technical Staff - Program Analysis This role is based in Palo Alto, California, and follows...  ...focused on our static analysis and ML‑for‑code initiatives. As an MTS, you...  ...Contribute to our program‑analysis infrastructure, which includes call graph construction... 

    Endor Labs

    Palo Alto, CA
    3 days ago
  • About the Role As a Member of Technical Staff [Platform] at NeoCognition , you’ll design and build the...  .... You’ll create the tooling, infrastructure, and developer experience that enable...  ...data pipelines , job scheduling , or ML experimentation workflows . Excellent... 

    NeoCognition Inc.

    Palo Alto, CA
    1 day ago
  • $180k

    Member of Technical Staff - RL Infrastructure About xAI xAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization... 
    Temporary work

    xAI

    Palo Alto, CA
    4 days ago
  • Member of Technical Staff — Diffusion Model About the Role RadixArk is seeking a Member of Technical...  ...Requirements 5+ years of experience in ML research or applied ML engineering...  ...generative AI About RadixArk RadixArk is an infrastructure-first AI company built by engineers... 
    Flexible hours

    RadixArk

    Palo Alto, CA
    4 days ago
  • RadixArk is hiring a Member of Technical Staff — CI Engineer to own the infrastructure that keeps SGLang moving. Our CI system runs 300+ GPU tests across NVIDIA, AMD,...  ...InfiniBand/RDMA experience in CI contexts Familiarity with ML inference workloads (model loading, KV cache,... 
    Flexible hours
    Night shift

    RadixArk

    Palo Alto, CA
    5 days ago
  •  ...Member Of Technical Staff, Software Engineer Inception creates the world's fastest, most efficient...  ..., platforms, tooling, and infrastructure that make every team at Inception more...  ...response. Collaborate across research, ML infrastructure, and product teams to... 
    Immediate start
    Flexible hours

    Inception LLC

    Palo Alto, CA
    2 days ago
  • $180k

     ...and retrieval algorithms that scale to millions of users Build and operate large scale search engine and search infrastructure Train state of the art ML ranking models Integrate Grok in search systems and products BASIC QUALIFICATIONS: Tech Stack: Python,... 
    Temporary work
    Shift work

    xAI

    Palo Alto, CA
    29 days ago
  •  ...real-world radiology datasets. About the Role As a Member of Technical Staff in Software Engineering, you will build the full...  ...-latency, and scalable. Collaborate closely with ML training, evaluation, and infrastructure teams to integrate inference into production workflows... 

    Cognita Imaging Inc.

    Palo Alto, CA
    2 days ago
  • $148.5k - $223.9k

     ...systems with customers. With your strong technical competence, strategic thinking and...  ...exceptional engineering skills. Has deep ML knowledge with meaningful implementation...  ...training, evaluation, and inference pipelines Infrastructure & Deployment Experience deploying ML... 

    Salesforce

    Palo Alto, CA
    5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Member of Technical Staff, ML Infrastructure. Be the first to apply!