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 3 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
    1 day ago
  •  ...hardware and robot systems to the infrastructure and state-of-the-art...  ...across levels — from senior to staff. What You'll Do Architect, build...  ..., distributed systems, ML infrastructure, or a closely...  ...candidates are expected to define technical direction and own architectural... 
    Suggested
    Immediate start

    Rhoda AI

    Palo Alto, CA
    1 day ago
  • $324k - $396k

     ...share knowledge with their teammates. Member of Technical Staff (X.AI LLC; Palo Alto, CA): Build...  ...post-baccalaureate experience in AI/ML research or development. * Alternatively...  ...robust, high-availability infrastructure for AI/ML services. ∙ Collaborating... 
    Suggested

    Xai

    Palo Alto, CA
    11 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...  ...software, distributed systems, compilers, ML systems, hardware‑software co‑design,... 
    Suggested

    DensityAI

    Mountain View, CA
    5 days ago
  •  ...beautifully efficient reality. You’re excited to design and support infrastructure not just for scale, but for speed, creativity, and discovery....  ...services). Experienced in designing infrastructure for ML workloads where performance, parallelism, and data movement are... 
    Suggested

    Odyssey

    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... 
    Full time
    Flexible hours

    RadixArk

    Palo Alto, CA
    2 days 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
    3 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
    2 days ago
  •  ...design custom ASICs alongside evolving ML workloads, and enable a new era of...  ...Intel. What You'll Do As a Founding Member of the Technical Staff on the RTL Design team at Architect,...  ...integration. Build and maintain RTL infrastructure for our in-house AI-driven flow: design... 

    Architect Labs

    Palo Alto, CA
    4 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
    H1b
    Work at office
    Work from home
    Work visa

    xAI

    Palo Alto, CA
    2 days ago
  •  ...custom ASICs alongside evolving ML workloads, and enable a new...  ...Intel. We're looking for staff/principal-level compiler...  .... What You’ll Do As a Member of the Technical Staff on the Compilers team...  ...Familiarity with MLIR or LLVM infrastructure. Leadership: Ability to lead... 

    Kindredventures

    Palo Alto, CA
    1 day ago
  • $148.5k - $223.9k

    Senior Member of Technical Staff - AI ResearchSkip to main content#Senior Member of Technical Staff...  ...exceptional engineering skills.** *Has deep ML knowledge with meaningful...  ...evaluation, and inference pipelines** *Infrastructure & Deployment** *Experience deploying... 
    Work at office

    Salesforce, Inc.

    Palo Alto, CA
    3 days 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
    14 hours 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
    14 hours 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
    2 days ago
  •  ...design custom ASICs alongside evolving ML workloads, and enable a new era of...  ...Intel. What You’ll Do As a Founding Member of the Technical Staff (Applied AI) at Architect, you’ll sit...  ...evaluation harnesses, and the domain‑specific infrastructure that makes agents useful on actual... 

    Architect Labs

    Palo Alto, CA
    14 hours 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
    4 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
    4 days ago
  • $190k - $250k

     ...Kubernetes, Slurm, Python, C++, PyTorch, and primarily on AWS. As an AI Infrastructure Engineer, you will be partnering closely with our Inference and...  ...monitoring, alerting, and observability solutions tailored to ML workloads running on Kubernetes and Slurm Respond swiftly to... 
    Full time

    Pantera Capital

    Palo Alto, CA
    3 days ago
  •  ...medals at the International Olympiads, and have published in top ML conferences. If your background doesn't fit cleanly...  ...quality in production. Search, Retrieval, Vector Systems & Data Infrastructure: Embedding pipelines, ranking and retrieval architecture, and... 

    Attention Engineering, Inc

    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
    1 day 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
    3 days ago
  •  ...healthcare. They’re developing advanced infrastructure that enables organisations to safely deploy...  ...with a strong engineering team on technically challenging problems. Design, build, and...  ...Alto) ⭐ Nice to Have Experience with AI/ML frameworks (e.g. PyTorch, TensorFlow) Strong... 
    Flexible hours
    3 days per week

    DeepRec.ai

    Palo Alto, CA
    14 hours ago
  • $148.5k - $223.9k

     ...workflows. Qualifications Exceptional engineering skills. Deep ML knowledge with meaningful implementation track records....  ...debugging model training, evaluation, and inference pipelines. Infrastructure & Deployment: experience deploying ML systems using Docker and... 

    Salesforce

    Palo Alto, CA
    14 hours ago
  • $180k

     ...teammates. ABOUT THE ROLE: We are seeking a highly skilled Senior Infrastructure Engineer to join our US Government Team, focused on designing...  ...of ownership, curiosity, and enthusiasm for tackling complex technical challenges in secure environments. Passion for problem‑... 
    Temporary work

    Pantera Capital

    Palo Alto, CA
    1 day ago
  • $230k

     ...users to effortlessly run large-scale ML applications, without the hassle of...  ...Systems Inc. has multiple openings for Sr. Member of Technical Staff. Title: Sr. Member of Technical...  ...detailed technical documentation for infrastructure configurations, inference workflows,... 
    Remote work

    Cerebras Systems

    Sunnyvale, CA
    3 days ago
  •  ...hardware that fuels it, recursively accelerating the path to artificial superintelligence. We are seeking a Software Engineer - Infrastructure (5+ years experience) to lead the productization and reliability of our core systems. You will architect the productization of... 

    Ricursive Intelligence

    Palo Alto, CA
    1 day ago
  •  .... Our goal is to co-design custom ASICs alongside evolving ML workloads, and enable a new era of domain‑specific chips that...  ...SuperIntelligence, xAI, Apple and Intel. What You’ll Do As Member of the Technical Staff - Software at Architect, you’ll build the core platform... 

    Architect Labs

    Palo Alto, CA
    4 hours ago
  • $324k - $396k

    Pantera Capital is seeking a Member of Technical Staff in Palo Alto, CA. This role involves introducing innovative techniques in AI, stabilizing large language model training, and working on hands-on technical problems. Candidates should have relevant experience and a Bachelor... 
    Remote job

    Pantera Capital

    Palo Alto, CA
    14 hours ago
  • $180k

    Member of Technical Staff, Recommendation Systems About xAI xAI’s mission is to create AI systems that can accurately understand the universe...  ...machine learning systems Who You Are Knowledge of data infrastructure like Kafka, Clickhouse, and Spark Experienced in implementing... 
    Temporary work
    Relocation

    xAI

    Palo Alto, CA
    3 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!