Extropic

The thrml package enables the generation and simulation of Ising and Boltzmann machines. It provides the necessary primitives to define thermodynamic state spaces and execute probabilistic sampling routines directly within JAX-based environments, facilitating research into non-von Neumann computational architectures.

This package is designed for research scientists, physicists, and machine learning engineers focused on probabilistic graphical models and thermodynamic computing. It is specifically intended for those building custom hardware-accelerated solvers for combinatorial optimization and sampling problems.

The primary dependency for thrml is JAX. Users must have a functional JAX environment configured to leverage the underlying acceleration primitives. It is designed to integrate into existing JAX-based research stacks without requiring additional proprietary runtime environments.