extropic-ai/thrml
Thermodynamic Hypergraphical Model Library in JAX
This is a specialized library for researchers and hardware engineers working with probabilistic graphical models, especially those exploring energy-based models. It helps simulate and prototype how these models would behave on new, energy-efficient computing hardware. You provide a definition of your probabilistic model, and it generates samples reflecting the model's underlying probabilities.
1,026 stars. Actively maintained with 1 commit in the last 30 days. Available on PyPI.
Use this if you are a researcher or hardware engineer developing or experimenting with novel hardware for probabilistic graphical models, particularly those that benefit from efficient block Gibbs sampling.
Not ideal if you are looking for a general-purpose machine learning library for tasks like image classification or natural language processing.
Stars
1,026
Forks
129
Language
Python
License
Apache-2.0
Category
Last pushed
Mar 11, 2026
Commits (30d)
1
Dependencies
2
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/extropic-ai/thrml"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/