Confusion

Generative conditional diffusion in JAX.

Installation

Install with

pip install git+https://github.com/limamau/confusion.git

to enable JAX in the GPU, it may be necessary to manually install it inside an environment with CUDA and GPUs available through

pip install -U "jax[cuda12]"