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
- ...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...SuggestedImmediate start
$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- ...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
- ...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
$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 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- ...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...
- ...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...
$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 workH1bWork at officeWork from homeWork visa- ...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...
$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- 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
- 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
- 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...
- ...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...
- 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
$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- ...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...
- 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
- ...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...
- ...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 hours3 days per week
$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...$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$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- ...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...
- .... 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...
$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$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 workRelocation
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Member of Technical Staff, ML Infrastructure. Be the first to apply!
- infrastructure support analyst Palo Alto, CA
- systems support technician Palo Alto, CA
- application support technician Palo Alto, CA
- desktop support analyst Palo Alto, CA
- tech aide Palo Alto, CA
- support technician Palo Alto, CA
- product support technician Palo Alto, CA
- helpdesk support technician Palo Alto, CA
- technical associate Palo Alto, CA
- help desk assistant Palo Alto, CA

