Extropic
Thermo ML Resident
$75K–$200K
Boston · On-site · Full-time · Posted Sep 8, 2026
- Experience
- No years specified; the residency is a minimum of 3 months.
- Visa sponsorship
- Offers are conditioned on authorization to receive controlled items under U.S. export control laws and regulations.
About the job
Develop and evaluate probabilistic machine-learning models and training methods for Extropic’s thermodynamic computing hardware.
What you'll do
- Derive theories for probabilistic models and learning rules with senior researchers.
- Build experimentation infrastructure and implement, visualize, and evaluate architectures, training algorithms, and benchmarks.
- Publish research, contribute to open source, and share design insights with the hardware team.
What they ask for
- Experience in scientific Python with JAX or a similar deep learning framework, such as PyTorch, TensorFlow, or Keras.
- Strong foundations in probability and linear algebra.
- Projects or papers demonstrating hands-on experience in applied machine learning and data science.
Nice to have
- Familiarity with deep learning theory and literature, including over-parameterization theory and scaling laws.
- Experience with energy-based or diffusion models, graph neural networks, deep learning infrastructure, information geometry, or computational Bayesian methods; publications at top ML conferences are also noted.
Pay
$75K – $200K; compensation type not specified
Tools
Python, JAX, PyTorch, TensorFlow, Keras, Slurm, Ray, Kubernetes
AI focus
Research and train probabilistic models, including energy-based and diffusion models, using a thermodynamic paradigm and specialized hardware.
Model-written summary · the original posting is authoritative