Avitai Bio

All

4 repositories

artifex
Public
A research-focused modular generative modeling library built on JAX/Flax NNX
research generative-adversarial-network vae
research generative-adversarial-network vae gans multi-modal normalizing-flows variational-autoencoder generative-models diffusion-model geometric-models
Python
•
MIT License
•0•1•0•0•Updated Mar 7, 2026Mar 7, 2026
datarax
Public
A Differentiable Data Pipeline Framework for JAX
data-science data machine-learning
data-science data machine-learning jit autograd data-analysis differentiable jax xla flax-nnx
Python
•
MIT License
•0•1•0•0•Updated Mar 6, 2026Mar 6, 2026
opifex
Public
A unified scientific machine learning framework built on JAX/Flax NNX
jax scientific-machine-learning sciml
jax scientific-machine-learning sciml flax-nnx
Python
•
MIT License
•0•2•0•5•Updated Mar 3, 2026Mar 3, 2026
calibrax
Public
Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical …
python benchmarking machine-learning
python benchmarking machine-learning deep-learning numpy ci-cd statistical-analysis scientific-computing flax profiling
Python
•
MIT License
•0•1•0•0•Updated Mar 3, 2026Mar 3, 2026