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 6 days ago
Similar jobs that could be interesting for youBased on the Research Engineer - ML Infrastructure in San Francisco, CA vacancy
- ...Oath Technologies is a new research organization building tools for... ...Role Summary Oath's research engineers will build novel formal... ...testing, agent development, AI/ML, and others that complement Oath... ...adjacent formal-verification infrastructure, with AI agents doing much of...SuggestedFlexible hours
Convergent Research
Berkeley, CA3 days ago $193.3k - $261.5k
...a key member of this team, you will lead research and development efforts in generative AI... ...innovation, and deep collaboration with engineering teams to bring research into production.Key... ...and optimization techniques- Knowledge of ML frameworks including JAX, PyTorch, vLLM,...SuggestedInternshipLocal areaFlexible hoursAmazonWebServices
San Francisco, CA2 days ago- ...teams for Google Workspace.Sierra’s Applied Research team advances the quality, speed, and... ...you’ll doTackle unsolved applied AI and ML problems. Pushing the boundaries of post-... ...Francisco, CAEmployment TypeFull timeLocation TypeOn-siteDepartmentEngineeringPlatform EngineeringSuggestedFull timeFlexible hours
Sierra
San Francisco, CA2 days ago $197.3k - $313.7k
...talented software and platform engineers to embed in our AI team to... ...directly enable world-class research and products used by millions... ...building substantive, scalable infrastructure is what enables scientific breakthroughs... ...and pipelines to support ML and AI efforts.Design, build,...SuggestedFull timeSalesforce
San Francisco, CA3 days ago$200k - $225k
About AlembicAlembic is where top engineers are solving marketing's hardest problem: proving... ...customers to leverage advanced analyticsDocument research and implementation decisions for... ..., or measurement domainsUnderstanding of ML engineering and MLOps practicesAbility to...SuggestedAlembic
San Francisco, CA4 days ago- ...About the Role Factory is seeking innovative Research Engineers to design and integrate advanced AI and ML capabilities that revolutionize productivity and accelerate innovation within software organizations.What you will do and achieveDesign, develop, and deploy AI-driven...Work at office
The San Francisco AI Factory
San Francisco, CA4 days ago - ...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...Full timeContract workShift work
SESAME
San Francisco, CA1 day ago - ...a year on administration instead of care. About the Role As a Research Engineer, you’ll be responsible for building industry-leading healthcare... ...Looking For 5+ years of experience with a minimum of 3+ in AI/ML engineering or research Prior experience post-training and...Work at officeFlexible hours
assorthealth
San Francisco, CA4 days ago - ...way it reinvented software engineering, and Chai is at the forefront... ...the RoleWe 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.Bachelor...Shift work
Chai Discovery
San Francisco, CA10 hours 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...Full timeWork at officeLocal areaRemote work
Invisible Technologies Inc. Defunct
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, CA4 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, CA1 day 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, CA1 day ago - ...intelligence layer to help scientists and engineers move faster and solve core R&D problems.... ...you’ll do: Translate cutting-edge AI research into production‑grade systems. Build evals... ...pipelines the next. Have applied AI/ML engineering and research experience,...
Altara
San Francisco, CA2 days ago $210k - $275k
...Research Engineer You'll build the evaluation systems that tell us whether Firecrawl actually... ...Full-Time Experience: 4+ years in ML, research engineering, or data-heavy backend... ...frontier: an AI company building the infrastructure other AI companies run on, not one...Full timeTemporary workVisa sponsorshipVisa Hunt
San Francisco, CA4 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....Full timeWork at officeRelocation packageBreakout Ventures
San Francisco, CA19 hours 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, CA1 day ago- ...Research EngineerLotus Health is a groundbreaking primary care app that integrates your medical... ....We are hiring a Founding Research Engineer to design, prototype, and ship the core reasoning... ...or JAXTrack record shipping applied ML or LLM features to real users, not just...
Lotus Health
San Francisco, CA1 day ago - ...is currently on-site) Industry: AI infrastructure / Reinforcement Learning (RL) training... ...Opportunity Our partner is hiring a Research Engineer to help scale the quality assurance (QA... ...of data quality, tooling, and applied ML operations: you’ll build the standards,...Remote work
talentpluto
San Francisco, CA19 hours 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, CA1 day ago- ...Job Description We are looking for a hybrid Systems Engineer and AI Researcher to lead the development of our agent evaluation framework... ...& RL Pipelines: Implement high-throughput post-training infrastructure (RLHF, Direct Preference Optimization, Process-Supervised...Work at office
Hyphen Connect
San Francisco, CA1 day ago $134k - $235k
...impact the world in a positive way. To learn more visit: As a Research Engineer in Neural Rendering, you will create the next generation of... ...concept. You are comfortable working with large datasets for ML applications. - Deep understanding of ML or neural rendering...Full timeWork at officeWork from homeFlexible hoursWaabi
San Francisco, CAa month ago$130k - $225k
...sensitive information in unexpected fields. Partner with engineering, research, operations, and customers to translate privacy requirements... ...At least 2 years of experience building production data or ML systems in Python, with strong Python proficiency. Hands-on...Full timeVisa sponsorshipClera
San Francisco, CA5 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, CA11 days ago - ...Description We are Genmo, a research lab developing the world’s most... ...an exceptional Software Engineer to join our research team in... ...Contributions to open-source ML projects Academic publications... ...papers Experience with ML infrastructure and deployment This position...Work at office
Genmo
San Francisco, CA26 days ago $211k - $251k
...learning and AI systems, with experience in areas such as applied ML, search, ranking, recommendations, advertising, language, or... ...datasets or high-volume ML products; or contributing to applied ML research or benchmarks. Responsibilities: Address unsolved applied...Full timeFlexible hoursSierra
San Francisco, CA5 days ago- ...in the physical world. Critical infrastructure is constrained by labor shortages, hazardous... .... About the Role At Watney, research engineers decide what our models learn to do... ...people deciding it. Write production ML code (Python, PyTorch, etc.). We’...
Watney
San Francisco, CA24 days ago - ...Chai Discovery seeks a Research Engineer to build agent-powered systems that automate the research... ...and evaluation Create datasets and infrastructure enabling agents to propose research... ...industry experience with agent-powered AI/ML systems Strong system design and architectural...
Chai Discovery
San Francisco, CA6 days ago $155k - $269k
...controllable, and efficient simulation. As a Research Engineer in the World Models team, you will... ...of the world model codebase and infrastructure. - Stay on top of emerging advances in... ...infrastructure and tooling for large-scale ML training (e.g., cloud platforms, Kubeflow...Full timeWork at officeWork from homeFlexible hoursWaabi
San Francisco, CAa month 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
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Research Engineer - ML Infrastructure. Be the first to apply!
Related searches
- deep learning research engineer San Francisco, CA
- junior machine learning research engineer San Francisco, CA
- research software engineer San Francisco, CA
- research engineer San Francisco, CA
- research programmer San Francisco, CA
- senior research engineer San Francisco, CA
- ai research engineer San Francisco, CA
- remote infrastructure engineer San Francisco, CA
- infrastructure engineer San Francisco, CA
- principal infrastructure engineer San Francisco, CA