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.
- 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...SuggestedFull timeRemote work
- ...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
$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...SuggestedTemporary work$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...SuggestedTemporary workH1bWork at officeWork from homeWork visa$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...SuggestedFull timeFlexible hours$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$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 workWork at officeWork from home- 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 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
- 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
$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 areaRelocation- ...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,...
- 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...
- ...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...
- ...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...
- 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...WorldwideFlexible hours
- 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...
- 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
- ...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...
- ...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...
- 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
$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...- 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...
$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- 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 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 hoursNight shift
- ...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 startFlexible hours
$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 workShift work- ...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...
$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...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Member of Technical Staff, ML Infrastructure. Be the first to apply!
- technical support assistant Palo Alto, CA
- technical analyst Palo Alto, CA
- end user support technician Palo Alto, CA
- IT assistant Palo Alto, CA
- help desk assistant Palo Alto, CA
- IT support technician Palo Alto, CA
- operations support technician Palo Alto, CA
- desktop support analyst Palo Alto, CA
- support analyst Palo Alto, CA
- technical associate Palo Alto, CA


