Machine Learning Research Engineer (MLRE) - Workflows/Systems
- Location
- San Francisco, CA (Remote) · Hybrid
- Job type
- Full-time
- Level
- Mid-level
- Experience
- 2+ years
- Listed
- 23 days ago
Skills & highlights for this role
- PyTorch
- JAX
- distributed computing
- asynchronous workflows
Also useful
- equivariant neural networks
- Flyte
- Dagster
About the Role
We're looking for a rare individual who thrives at the intersection of machine learning systems architecture and distributed computing. You will help architect the future of molecular machine learning by enabling our scientific teams to flexibly conduct experiments at scale, pushing the boundaries of foundation simulation models.
While we prefer candidates willing to work from our San Francisco office, highly skilled candidates may be considered for working from New York City with travel to San Francisco as needed. Both locations are offered as hybrid roles, spending at least some of your time working from the office in collaboration with coworkers. Travel is part of all roles at Achira, both to conferences and corporate on-site activities.
What You’ll Do
- Build and maintain robust multi-stage asynchronous workflows for running data generation, training, and evaluations for our machine learning stack.
- Rationalize machine learning systems design and software architecture.
- Identify blockers and build solutions that scale to the size of foundation models.
- Operate as the glue between research scientists and the infrastructure team.
About You
- At least two years relevant industry experience.
- Highly fluent in and enthusiastic about PyTorch and JAX.
- Used to thinking in asynchronous primitives.
- Strong views on library design: clean abstractions, minimal surface area, consistency.
- Solid track record of observable artifacts (e.g., GitHub) showing clear, well-documented code.
- ML generalist who knows what scalable, reliable ML systems look like.
Nice to Have
Even if you hit none of these bonus features, we encourage you to apply!
- Experience with equivariant architectures, geometric deep learning, or GNNs (NequIP, MACE, SchNet, PaiNN, or similar), and/or ML-assisted drug discovery.
- Experience building in declarative workflow orchestration frameworks like Flyte, Dagster, etc.
- Lack of fear around interacting with quantum chemical scientists and their data pipelines.
Why Achira
At Achira, we are building a team of world-class scientists, ML researchers, and engineers to work together to move beyond the beaten path in drug discovery. We are actively exploring the next frontier of model architectures for AI x Chemistry: developing world models for the physical microcosm. Our goal is to make biology at the molecular level something that can be learned, predicted, and designed.
At Achira, you’ll operate at the frontier scale of massive compute, massive data, and massive ambition. You’ll own impactful work end-to-end, from ideation to architecture to deployment on distributed infrastructure. We are a well-funded, talent-dense organization that values rigor, speed, execution, and an ownership mindset. We’re looking for new members who share our sense of relentless urgency and are natural collaborators who value team success.
From Achira's careers page, first listed on Rolev.ai on October 17, 2026. Rolev.ai is not the employer.
How Rolev.ai helps with this application
Rolev.ai is a job search and application tool for US job seekers. For this Machine Learning Research Engineer (MLRE) - Workflows/Systems role at Achira, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.
See how you fit
Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.
Tailor without inventing
Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.
Draft the full kit
Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.
1-Click Extension Autofill
The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.
Similar roles
All Machine Learning Engineer jobs- Senior / Staff Machine Learning Engineer, ACAI Training & Inference PlatformApple · Cupertino
- Machine Learning Engineer 5 (Senior Manager, IC)Capital One · Chicago, IL
- Research, Tinker, RL SystemsThinking Machines Lab · San Francisco, CA
- Machine Learning Engineer, Answers RelevanceReddit · Remote -
- Senior ML Engineer, Core DevelopmentAnduril · Costa Mesa, CA
- Machine Learning Engineer (TS/SCI)Maxar · Herndon, VA
Common questions
Where is the Machine Learning Research Engineer (MLRE) - Workflows/Systems job at Achira located?
Yes, this Machine Learning Research Engineer (MLRE) - Workflows/Systems opening at Achira is listed as San Francisco, CA (Remote) (Hybrid). Always check Achira's application page for specific state eligibility or relocation policies.
What qualifications are required for Machine Learning Research Engineer (MLRE) - Workflows/Systems at Achira?
For the Machine Learning Research Engineer (MLRE) - Workflows/Systems role, Achira highlights 2+ years of relevant experience; experience with PyTorch, JAX, distributed computing, asynchronous workflows. Rolev.ai can score your existing resume against these requirements and show any skill gaps before you apply.
How do I apply for the Machine Learning Research Engineer (MLRE) - Workflows/Systems position at Achira?
This listing links directly to Achira's Ashby application page. You can apply directly on Achira's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Ashby application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.