Achira
ML Research Scientist (MLRS) - Generative AI
San Francisco Office
Sponsorship not specifiedDetected 21 days ago
PythonMachine LearningDeep LearningPyTorchLLMsResearchCommunicationCollaboration
About the role
- We're looking for machine learning researchers who want to shape the frontier of generative models for the atomistic microcosm.
- You will work at the intersection of cutting-edge machine learning, statistical mechanics, and approximate bayesian inference to help us conquer sampling and generation problems at light-speed.
- 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.
Responsibilities
- Design and train frontier generative models: diffusion, autoregressive, flow-based, and latent-variable architectures.
- Build models that can map between data distributions to bridge the gap between simulation and reality.
- Collaborate with physicists and chemists to ensure models are grounded in real physics.
- Work with research engineers and the infrastructure team to identify where research ideas will need support in order to deliver effective results.
- Interested in building generative models that describe real matter.
- Drive to build at the frontier of what's possible and try out new, high-risk ideas.
Requirements
- Strong interdisciplinary communication and presentation skills and the ability to translate ideas and concepts to colleagues from non-ML backgrounds.
- Proficiency in Python and modern ML frameworks (PyTorch, JAX).
- Experience collaborating on research projects across multi-person teams.
Nice to have
- Achira values excellent ML researchers from many backgrounds, and expect members of the team to contribute complementary strengths.
- Experience working with models that operate on 3-D point clouds and dynamic data.
- Experience in sequential monte carlo methods.
- Experience with probabilistic programming.
- Familiarity with statistical mechanics: working knowledge of sampling, estimators, and the Crooks/Jarzynski perspective of nonequilibrium statistical mechanics.
- Prior experience working in or with researchers in the domains of computational chemistry, biology, or materials science.
- Experience working with multi-cloud distributed compute systems.
- Experience working with multi-site distributed company team.
Benefits
- Machine learning researcher with professional experience (post-degree) in an industry setting.
- Demonstrated research impact through conference talks or publications (in machine learning venues), open-source contributions, or released models.
This listing is sourced directly from Achira's careers page and normalized into a canonical job model.