avitai-artifex
Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.
View on Github
Activity
- Latest release
- 4d ago
- Total releases
- 7
- Cadence
- ~17 days
- Last 12 months
- 7
Reach
- Stars
- 1
Details
- License
- custom
- First release
- Apr 25, 2026
Releases
| Version | Released | |
|---|---|---|
0.1.7
patch
| ||
0.1.6
patch
| ||
0.1.5
patch
| ||
0.1.4
patch
| ||
0.1.3
patch
| ||
0.1.2
patch
| ||
0.1.0
initial
|