ML Research Engineer - Research

Achira • San Francisco, CA

20h • Hybrid

About The Position

We're looking for a rare individual who thrives at the intersection of applied machine learning research and rigorous software engineering. You will advance the state of the art in foundation simulation models by implementing and experimenting with internal and literature-sourced ideas, participating with research teams to scale our ML systems, train and evaluate models, and engineer scientific prototypes into production. 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.

Requirements

• Strong software engineering fundamentals, with experience not just building one-off scripts but reproducible pipelines for research, writing necessary documentation, and observing coding best-practices.
• Track record of observable artifacts (e.g., GitHub, papers) showing work in ML or scientific computing libraries.
• Solid working knowledge of PyTorch and JAX and the modern ML research stack.
• Comfortable with HPC or large-scale compute environments, and used to thinking on the scale of hundreds or thousands (or even more!) fits running at once.
• Sufficient scientific depth to engage with the research questions, whether developed through prior industry experience or during a PhD.

Nice To Haves

• Experience with equivariant architectures, geometric deep learning, or GNNs (NequIP, MACE, SchNet, PaiNN, or similar).
• Familiarity with generative modeling: diffusion models, flow matching, score-based methods.
• Regular involvement in open-source ML or scientific computing libraries.
• Experience building agent-driven research, active learning, and data curation pipelines.

Responsibilities

• Design and run experiments to test out hypotheses on the path to foundation model development.
• Engineer meaningful evals and metrics which enable rapid model iteration.
• Design, build and maintain scalable, reproducible libraries for training, experimentation evaluation, and simulation, in service of large-scale research initiatives.
• Implement model architectures both from the literature and developed in collaboration with our in-house researchers that push the boundaries of molecular simulation.
• Enable agent-driven research and workflows and maintain guardrails on agentic tooling.
• Help prepare manuscripts, software artifacts, and datasets for public release.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: Ph.D. or professional degree