Machine Learning Engineer

Why AchiraAt 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.About the RoleWe'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.What You'll DoDesign 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.About YouStrong 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 HaveEven 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).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.