antonioterpin / jkonet-starView on GitHub
JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
38Mar 18, 2025Updated last year

Alternatives and similar repositories for jkonet-star

Users that are interested in jkonet-star are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?