ARDM

github.comsite

Official implementation of Autoregressive Diffusion Models.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   
2. ▲
   D3PM github.com

   site category #jax link ↗

   Official implementation of Structured Denoising Diffusion Models in Discrete State-Spaces.

3. ▲
   Understanding Autodiff with JAX by Srihari Radhakrishna radx.in

   site category #jax link ↗

   Understand how autodiff works using JAX.

4. ▲
   Advanced R, 2nd ed. by Hadley Wickham (2019) adv-r.hadley.nz

   site category #r link ↗
5. ▲
   Jraph github.com

   tool category #python-data-science link ↗

   A Graph Neural Network Library in Jax.

   
6. ▲
   JAX, M.D. github.com

   tool category #jax link ↗

   Accelerated, differential molecular dynamics.

   

JAX › Models and Projects > Flax: “Official implementation of Autoregressive Diffusion Models.”