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
- 1mo ago
- Total releases
- 15
- Cadence
- ~daily
- Last 12 months
- 15
Reach
- Stars
- 1
Details
- License
- custom
- First release
- Apr 25, 2026
Releases
| Version | Released | |
|---|---|---|
0.1.15
patch
| ||
0.1.14
patch
| ||
0.1.13
patch
| ||
0.1.12
patch
| ||
0.1.11
patch
| ||
0.1.10
patch
| ||
0.1.9
patch
| ||
0.1.8
patch
| ||
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
|