extropic-ai / thrml
Thermodynamic Hypergraphical Model Library in JAX
Why this job: works in the areas we cover, but does not say which job it does
What it does
THRML is a JAX library for building and sampling probabilistic graphical models, with a focus on efficient block Gibbs sampling and energy-based models. Extropic is developing hardware to make sampling from certain classes of discrete PGMs massively more energy efficient; THRML provides GPU‑accelerated tools for block sampling on sparse, heterogeneous graphs, making it a natural place to prototype today and experiment with future Extropic hardware.
Get started
pip install thrml- Maintenance
- Active
- Last commit 2 days ago
- Stars
- 1,174
- +3 this week
- Licence
- Apache-2.0
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- v0.1.4
- Released 3 months ago
- Built with
- Python
- 65% of the code
Is it looked after?
- Contributors
- 6
- Last 90 days
- Issues
- 47% closed
- 8 still open
- Pull requests
- 38 merged
- 7 waiting
- Releases
- 2
- 1 in the last year
- Contributing guide (opens in new tab)
- No security policy
- No code of conduct
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| NX-AI / xlstm-jax | 105 | 6 days ago | Custom | No downloads |
More: languages, topics and project record
Languages
- Python 65%
- HTML 24%
- CSS 10%
- JavaScript 1%
Topics
- jax
- machine-learning
- probabilistic-computing
- probabilistic-graphical-models
Project record
- JobOther AI tools
- StatusListed
- Commits71
- Forks144
- Watching28
- Stars in 28 days+10
- CreatedSep 16, 2025
- First seen by usNov 7, 2026
- CheckedNov 12, 2026