Machine Learning Research Engineer (MLRE) - Research

Sourced from feeny.ai

Achira is hiring a Machine Learning Research Engineer to advance foundation simulation models at the intersection of AI and chemistry. You will design and run experiments, build scalable reproducible training and evaluation libraries, and implement novel model architectures for molecular simulation. The role blends applied ML research with rigorous software engineering on large-scale distributed infrastructure.

• Design and run experiments to test hypotheses toward foundation model development
• Engineer meaningful evals and metrics to enable rapid model iteration
• Design, build, and maintain scalable, reproducible libraries for training, evaluation, and simulation
• Implement model architectures from literature and in collaboration with in-house researchers
• Enable agent-driven research workflows and maintain guardrails on agentic tooling
• Help prepare manuscripts, software artifacts, and datasets for public release

• Strong software engineering fundamentals with reproducible research pipelines and best practices
• Track record of observable artifacts (GitHub, papers) in ML or scientific computing
• Solid working knowledge of PyTorch and JAX and the modern ML research stack
• Comfortable with HPC or large-scale compute environments running many jobs at once
• Sufficient scientific depth to engage with research questions via industry experience or PhD

• Experience with equivariant architectures, geometric deep learning, or GNNs (NequIP, MACE, SchNet, PaiNN)
• 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