> Markdown version of [/jobs/ext/2966163-machine-learning-research-engineer](https://www.wearedevelopers.com/jobs/ext/2966163-machine-learning-research-engineer). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Machine Learning Research Engineer

- **Company:** Achira Inc.
- **Location:** San Francisco, CA, United States
- **Experience:** Experienced
- **Contract:** Permanent contract
- **Skills:** Airflow, Distributed Systems, Github, Machine Learning, Software Architecture, Pytorch, Machine Learning Operations, Data Pipelines
- **Published:** September 17, 2026
- **Apply:** https://startup.jobs/machine-learning-research-engineer-mlre-workflows-systems-achira-8134382

## About the Role

* 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., * 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.

## Description

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., * 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.