Latest AI JobsBrowse Jobs
E

Thermo ML Resident

extropicSan FranciscoApplications & Algorithms
Apply on company site
Salary
$75K – $200K
Job type
Full-time
Experience
Entry
Posted
8 Sept 2026
Status
Open · checked 7d ago

About this role

San Francisco is the location for the Thermo ML Resident position at Extropic, offering a flexible part-time or full-time residency for junior scientists interested in probabilistic inference hardware. This program suits developing machine learning professionals seeking substantial autonomy over a minimum duration of three months.

Role Overview

Extropic operates a flexible residency program designed for junior machine learning scientists to pioneer model training within a thermodynamic hardware paradigm. Participants receive high levels of autonomy while working on specialized hardware that accelerates specific probabilistic inference tasks.

Key Responsibilities

Work alongside senior researchers to formulate the theory behind novel probabilistic models and learning rules, such as diffusion and energy-based models
Expand experimentation infrastructure while optimizing model design configurations
Build, inspect, and assess fresh architectures, training algorithms, and evaluation benchmarks
Share design findings with the hardware group, release open-source contributions, and author academic papers

Requirements

Background using scientific Python alongside deep learning frameworks like JAX, PyTorch, TensorFlow, or Keras
Solid grounding in linear algebra and probability principles
Completed projects or published papers showing practical data science and applied machine learning capabilities

Nice to Have

Understanding of deep learning literature and theory, covering scaling laws and over-parameterization concepts
Background in training diffusion models or energy-based models
Working knowledge of graph neural networks or related message passing algorithms
Familiarity with training and experimentation infrastructure such as Slurm, Ray, Kubernetes, and Weights & Biases
Theoretical knowledge regarding information geometry
Understanding of computational Bayesian approaches including variational inference and MCMC sampling
Academic papers published in major machine learning venues like NeurIPS, ICML, ICLR, or CVPR

Skills and Tools

Python
JAX
PyTorch
TensorFlow
Keras
Slurm
Ray
Kubernetes
Weights & Biases

Salary and Benefits

The salary range for this role is between $75K and $200K.

About the Company

Extropic functions as an equal opportunity employer and requires employment offers to be contingent upon compliance with U.S. export control laws and regulations.

How to Apply

Interested candidates should apply directly on the official company careers page.

Skills and tech stack

Questions candidates ask

What is the duration of the residency program?

The minimum duration is 3 months, with options for part-time or full-time schedules.

Where is the job located?

The position is located in San Francisco.

What is the salary range for the role?

The compensation ranges from $75K to $200K.

What technical framework experience is required?

Candidates need experience in scientific Python using JAX, PyTorch, TensorFlow, or Keras.

Are there any legal or compliance requirements for the job?

Yes, employment is conditional on authorization to receive items controlled under U.S. export control laws and regulations.

Ready to apply?

The application is on the official extropic careers page.

Apply on company site