Thermo ML Resident
Extropic, Boston
About the job
Overview
Extropic is looking for junior ML scientists to join our residency program on either a part-time or full-time basis. This is a flexible program that can be similar to an internship (minimum 3 months) but we give our residents far more autonomy than most internships.
Our hardware massively accelerates certain kinds of probabilistic inference, and residents will help pioneer the science of training models in the thermodynamic paradigm.
Responsibilities
- Collaborate with senior researchers to derive the theory of new probabilistic models and their learning rules, including energy-based models and diffusion models
- 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
Required Qualifications
- Experience in scientific Python with JAX or similar deep learning framework (PyTorch, TensorFlow, or Keras)
- Strong foundations in probability and linear algebra