Research Engineer, ML Infrastructure
Chai Discovery
Chai Discovery seeks a Research Engineer to optimize machine learning infrastructure for molecular design models. The role involves building training systems, profiling GPU clusters, and ensuring scalability and reliability for large-scale computational workloads supporting drug discovery.
Responsibilities
- Develop and optimize training stack across model, layer, and kernel levels
- Profile end-to-end training on large GPU clusters; identify and eliminate bottlenecks
- Ensure new model architectures and training recipes scale from experiments to frontier-scale runs
- Implement fault tolerance, checkpointing, and deterministic orchestration for long-running jobs
Requirements
- Deep industry experience with AI/ML teams on large-scale systems
- Strong software systems design and architecture skills
- Proficiency in Python programming
- Expertise in PyTorch or JAX frameworks
- Demonstrated impact on real-world ML systems and problems
Vacancy posted a month ago
Similar jobs that could be interesting for youBased on the Research Engineer, ML Infrastructure in San Francisco, CA vacancy
- ...Research EngineerSesame believes in a future where computers are lifelike - with the ability... ...all of our product goals. As a Research engineer at Sesame, you are a research-oriented person... ....Proven software engineer who loves ML; comfortable writing production code across...SuggestedFull timeContract workShift work
SESAME
San Francisco, CA4 days ago $180k - $250k
...Research EngineerSuperAnnotate helps the world's leading AI teams... ...engagements that require real ML depth, and emerging opportunities... ...of the field. As a Research Engineer, you'll take a research... ...eval harnesses, environments, infrastructure – without relying on a platform...SuggestedFull timeSuperAnnotate AI
San Francisco, CA3 days ago- ...Research EngineerNew York - Hybrid; San Francisco Bay Area - HybridAbout... ...clients. As a Research Engineer, you'll work on how frontier... ...client requirements, and with ML software engineers who build... ...modern ML stackEvaluation & Infrastructure: Real experience building evaluation...SuggestedFull timeWork at officeLocal areaRemote work
Invisible Technologies Inc. Defunct
San Francisco, CA4 days ago - ...Role Description This is a Research Engineer role focused on post‑training and reasoning. Full‑time or part‑time; we're open to candidates currently... ...and interpretability. What we're looking for: Strong ML engineering fundamentals. Comfortable training and fine‑tuning...SuggestedFull timePart timeRemote workFlexible hours
Cobalt.io
San Francisco, CA2 days ago $180k - $250k
...Open role Research Engineer San Francisco (On-site), Full-time About Us Constellation is creating the AI-human... ...keep exploring. Qualifications ~3+ years building ML systems or research infrastructure, including distributed training runs on 100+ GPUs....SuggestedFull timeWork at officeRelocation packageBreakout Ventures
San Francisco, CA3 days ago- ...you want to help change that, apply below.About The RoleAs a Research Engineer, you'll be responsible for building industry-leading healthcare... ...Looking For5+ years of experience with a minimum of 3+ in AI/ML engineering or researchPrior experience post-training and deploying...Work at office
Assort Health
San Francisco, CA3 days ago - ...prescriptions. Our team includes ex-founders and engineers who have built and scaled consumer apps... ...Overview We are hiring a Founding Research Engineer to design, prototype, and ship... ...or JAX Track record shipping applied ML or LLM features to real users, not just...
Lotus
San Francisco, CA3 days ago $200k - $400k
...Research EngineerDecagon is the leading conversational AI platform empowering every brand... ...grow as a team.About the RoleAs a Research Engineer, you'll be responsible for building industry... ...Like This2+ years of experience in AI/ML engineering or research.Prior experience...Full timeWork at officeLocal areaDecagon
San Francisco, CA4 days ago- ...PhonicPhonic is a product and research lab focused on powering the... ....About The RoleAs a Research Engineer at Phonic, you'll sit at the intersection of cutting-edge ML research and production engineering... ...), while also building the infrastructure and tooling needed to move...Work at office
Phonic
San Francisco, CA4 days ago $9.7k - $19k
...Research Engineer InternThe Center for AI Safety (CAIS) is a leading research and advocacy organization focused on mitigating societal-scale... ...level.Have co-authored at least one paper published at a top ML conference venue (e.g., NeurIPS, ICML, ICLR, ACL, CVPR). Workshop...Full timeInternshipLocal areaCenter for AI Safety
San Francisco, CA4 days ago- ...San Francisco, CA, US Full-time Will sponsor 3+ years Machine Learning We’re hiring a Founding ML Research Engineer to work work through the full stack across data, pre-training, post-training & evals for training Generalist Audio Models. You’ll work through the entire...Full time
Kalpa Labs
San Francisco, CA2 days ago $140k - $250k
...Industry: AI training data infrastructure Compensation: $140K-$250K... ...priority and a genuinely hard research problem. Because data flows through... ...to growth. As a Research Engineer, you will build the automated... ...field Background in AI/ML engineering, or software engineering...Full timeWork at officetalentpluto
San Francisco, CA4 days ago- ...way it reinvented software engineering, and Chai is at the forefront... ...Role We are seeking an AI Research Engineer to help design,... ...optimize Chai's core models and infrastructure, working in tight loops with... ...and GPU programming, AI/ML + data systems at scale....Shift work
Chai Discovery
San Francisco, CA1 day ago - ...AfterQuery is an applied research lab curating data solutions... ...will own and architect core infrastructure systems that power our platform... ...the opportunity to shape the engineering organization and lead major... ...engineering, research engineering, ML infrastructure, or a closely...Local area
AfterQuery
San Francisco, CA1 day ago $120k - $200k
...We are actively seeking a Research Engineer specializing in Machine Learning and AI to play a pivotal role in pioneering advanced solutions.... ...projects with end-to-end responsibility. Deep understanding of ML and AI, focusing on building scalable solutions for large-...Casual workWork at officeErth.AI Inc.
San Francisco, CA2 days ago$350k
...Research Engineer / Scientist, AlignmentSan Francisco, CAAbout AnthropicAnthropic's mission is to create reliable, interpretable, and steerable... ...Policy.You May Be a Good Fit If YouHave significant software, ML, or research engineering experienceHave some experience...Work at officeVisa sponsorshipFlexible hoursAnthropic
San Francisco, CA4 days ago- ...Francisco, California. The Role: As a Research Engineer - AI Performance & Kernel Optimization... ...and optimization for large-scale ML workloads, using any level of the stack... ...55x Collaboration with research and infrastructure teams to turn systems improvements into...Work at officeRelocation package
Zyphra
San Francisco, CA29 days ago - Freenome seeks a Senior Bioinformatics Research Engineer to advance our computational platform for R&D, building and deploying bioinformatics software and tooling in collaboration with biologists, ML scientists, and software engineers. The role reports to the Director...Remote job
- ...Perception AI Engineer Specter's mission is to help automate the physical world. Today, we build video sensors with state-of-the-art AI agents that answer any question, anywhere in their environments. Our systems can automatically detect and reason about any physical...
Specter Services LLC
San Francisco, CA1 day ago - ...expertise in the models you ship. We're looking for strong CV/ML engineers with good product judgment who are comfortable both designing... ...scanned, handwritten, imperfect Practical mindset - you read the research, but your focus is on shipping working code Comfortable in...For contractorsWork at office
Bobyard, Inc
San Francisco, CA4 days ago - Achira is seeking a talented individual for a hybrid role focused on applied machine learning research and software engineering in San Francisco. You will design and run experiments, engineer metrics, and maintain libraries for large-scale research initiatives. Ideal candidates...
- Machine Learning Research Engineer San Francisco · Onsite · Full-time About Bracket Bot We believe... ..., software and operating system; the infrastructure and distributed systems they connect to... ...source work, research projects, or shipped ML systems that demonstrate your approach....Full timeLocal area
- ...We're building a research cohort of senior domain experts in ML inference and GPU kernel engineering — spanning inference serving systems, kernel optimization, and large-model deployment infrastructure — to help evaluate how frontier AI models reason about hard, real-...Temporary work
AfterQuery
San Francisco, CA15 days ago - ...enough to ship or hand off to engineering Run experiments, read the... ..., modeling, and serving infrastructure Builds with AI as a natural... ...technical field, or equivalent research experience (approximately 2... ...experience building ML systems that shipped and ran...
Jobrapido
San Francisco, CA1 day ago - ...AI Research Engineer OpportunityPoly is building a better file storage platform for everyone. We've raised $8M from YC, Bloomberg Beta, Felicis... ..., including the ability to write down or describe core ML modeling equations such as gradient descent, scaled dot product...Work at office
Poly
San Francisco, CA3 days ago - ...on the AI brains behind The Path’s AI Therapist. Combine research, data science, and engineering to create models, orchestration, and evaluation systems... ...maintain in the product. Competencies # LLM and Applied ML Depth. Demonstrates strong experience with large...
Mental, LLC
San Francisco, CA2 days ago $100k - $300k
...cutting edge, we blend frontier research with real-world execution.... ...s mission. As an AI Research Engineer, you will work at the frontier... ...mission to make the world's infrastructure safer. What You'll Do & Achieve... ...cutting-edge developments in ML and program analysis, and...Cogent Security
San Francisco, CA2 days ago- ...Machine Learning Engineer For Robot Dexterous Manipulation You... ...learn more effectively. Our research involves vision, motion, biomechanical... ..., training, and evaluation infrastructure to support rapid... ...training, distributed computing, or ML infrastructure is a plus....Full time
Relari
San Francisco, CA2 days ago $214k - $285k
...AI Research Engineer Hex is an AI-powered platform for modern Data Science and Data Analytics... ...experiments, fine-tune models, deploy AI infrastructure, and build and maintain experimentation... ...you have: Experience getting AI/ML capabilities into production and...Full timeWork at officeFlexible hoursHEX
San Francisco, CA23 hours ago$122k - $215k
...Research EngineerWaabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we're unlocking... ...candidates who want to impact the world in a positive way.As a Research Engineer, you will be at the forefront of advancing and deploying...Full timeWork at officeWork from homeFlexible hoursWaabi
San Francisco, CA4 days ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Research Engineer, ML Infrastructure. Be the first to apply!
Related searches
- research engineer San Francisco, CA
- research programmer San Francisco, CA
- deep learning research engineer San Francisco, CA
- ai research engineer San Francisco, CA
- research software engineer San Francisco, CA
- data infrastructure engineer San Francisco, CA
- remote infrastructure engineer San Francisco, CA
- infrastructure engineer San Francisco, CA
- infrastructure engineering manager San Francisco, CA
- principal infrastructure engineer San Francisco, CA