The role involves leading research in probabilistic ML, collaborating with other experts, scaling experimentation, and deploying models. Candidates should have a strong research background and publications in top ML conferences.
Overview
Extropic’s hardware massively accelerates certain kinds of probabilistic inference. 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 properties, and deploy performant models. 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.
Responsibilities
- 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.
- 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.
Required Qualifications
- Experience in scientific Python and at least one deep learning framework (PyTorch, JAX, TensorFlow, Keras)
- Extremely strong foundations in probability and linear algebra
- Familiarity with deep learning theory and literature, including theory of over-parameterization and scaling laws
- Publications in top ML conferences (NeurIPS, ICML, ICLR, CVPR)
- Experience training high-performance models, including familiarity with infrastructure (Slurm, Ray, Weights & Biases)
- Experience deploying models, including familiarity with infrastructure (Ray, AWS, ONNX)
Preferred Qualifications
- 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
Extropic is an equal opportunity employer
Top Skills
AWS
Jax
Keras
Onnx
PyTorch
Ray
Ray
Scientific Python
Slurm
TensorFlow
Weights & Biases
Similar Jobs
Machine Learning • Security • Software • Analytics • Defense
Lead the development of disruptive AI/ML algorithms for signals exploitation and system resource management, guiding project teams and communicating with customers.
Top Skills:
C/C++JaxMatplotlibPandasPythonPyTorchScikit-LearnTensorFlow
Machine Learning • Security • Software • Analytics • Defense
As a Lead Research Scientist in Machine Learning, you will develop AI/ML algorithms, lead teams, work on innovative problems, and communicate technical insights.
Top Skills:
C/C++JaxMatplotlibPandasPythonPyTorchScikit-LearnTensorFlow
Machine Learning • Security • Software • Analytics • Defense
Develop AI/ML algorithms for signals exploitation, perform modeling and simulations, and engage in project teams for innovative solutions.
Top Skills:
C/C++JaxMatlabMatplotlibPandasPythonPyTorchScikit-LearnTensorFlow
What you need to know about the Boston Tech Scene
Boston is a powerhouse for technology innovation thanks to world-class research universities like MIT and Harvard and a robust pipeline of venture capital investment. Host to the first telephone call and one of the first general-purpose computers ever put into use, Boston is now a hub for biotechnology, robotics and artificial intelligence — though it’s also home to several B2B software giants. So it’s no surprise that the city consistently ranks among the greatest startup ecosystems in the world.
Key Facts About Boston Tech
- Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
- Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
- Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
- Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories
