hotjax

Probabilistic Machine Learning with Energy-Based Models in JAX


Keywords
deep-learning, density-estimation, energy-based-model, jax, machine-learning, probabilistic-machine-learning, probabilistic-models
License
Unlicense
Install
pip install hotjax==0.0.1

Documentation

HotJAX | Probabilistic Machine Learning with Energy-Based Models in JAX

codecov CI

Install it from PyPI

pip install hotjax

Usage

from hotjax import BaseClass
from hotjax import base_function

BaseClass().base_method()
base_function()

Development

Read the CONTRIBUTING.md file.