Extropic
Research Scientist - Machine Learning
San Francisco
Sponsorship not specifiedDetected 31 days ago
PythonAlgorithmsAWSMachine LearningDeep LearningTensorFlowPyTorchA/B TestingResearch
> stay_score
odds of building a lasting career here
16Unrated
Cap-exempt (no lottery)0
Sponsors this role0
Entry-level history0
PERM / green-card track0
Lottery odds40
Fits your clock70
No strong sponsorship signal in the public record yet. In the full product we resolve the exact legal entity and show its filing history with a confidence score — treat as unverified until then.
Lottery odds assume a STEM candidate.
Personalize to your clock →> community_outcomes
No reports yet — be the first to help the next applicant.
About the role
- Extropic's hardware massively accelerates certain kinds of probabilistic inference.
Responsibilities
- Scale up experimentation infrastructure and optimize over the design space of models
- Implement, visualize, and evaluate new architectures, training algorithms, and benchmarks
- Publish papers, contribute to open source, and communicate design insights to our hardware team
- Create production models for domain experts using customer data
- Senior hires will be leading their own research direction and are therefore expected to quickly become experts across our abstraction stack, including the hardware, software, physics, and math.
Requirements
- Experience training high-performance models, including familiarity with infrastructure (Slurm, Ray, Weights & Biases)
- Experience deploying models, including familiarity with infrastructure (Ray, AWS, ONNX)
Nice to have
- Experience designing probabilistic graphical models (PGM)
- Experience training energy-based models (EBMs) or diffusion models
- Experience with numerical methods in diffeq solvers
- Experience with message passing or training graph neural networks (GNNs)
- Strong theoretical background in information geometry
- Strong theoretical background in random matrix theory
- Strong grasp of computational Bayesian methods, including MCMC sampling methods and variational inference
Benefits
- Collaborate with senior researchers, residents, engineers, and physicists to derive the theory of new probabilistic models and their learning rules, including energy-based models and diffusion models
- Experience in scientific Python and at least one deep learning framework (PyTorch, JAX, TensorFlow, Keras)
- Familiarity with deep learning theory and literature, including theory of over-parameterization and scaling laws
Company info
- Our ML team works on the science of training models in the thermodynamic paradigm, and we are looking for senior research and engineering talent to derive probabilistic ML theory, empirically demonstrate its scaling, and deploy performant models.
Apply directly at Extropic →Create a free account for alerts like thisView Extropic immigration profile
This listing is sourced directly from Extropic's careers page and normalized into a canonical job model.