Pinned Loading

1. avitai/datarax avitai/datarax Public

   Differentiable data pipelines for JAX/Flax NNX: HuggingFace, TFDS and ArrayRecord sources, augmentation stages, scan-based epochs, exact mid-epoch resume.

   Python 3

2. avitai/artifex avitai/artifex Public

   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; ima…

   Python 1

3. avitai/opifex avitai/opifex Public

   Scientific machine learning for JAX/Flax NNX: neural operators (FNO family, DeepONet, PINO, UNO), physics-informed networks (PINN, FBPINN, XPINN), E(3)-equivariant atomistic potentials (SchNet, Pai…

   Python 3 1

4. avitai/calibrax avitai/calibrax Public

   JAX benchmarking, profiling and evaluation metrics for Flax NNX: a registry of pure-function metrics (regression, classification, calibration, uncertainty, forecasting, generative, image, text, aud…

   Python 2

5. avitai/DiffBio avitai/DiffBio Public

   End-to-end differentiable bioinformatics for JAX/Flax NNX: soft Smith-Waterman alignment, read mapping and assembly, variant calling, RNA-seq, single-cell, epigenomics, CRISPR, metabolomics, multi-…

   Python 2