GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
created at Aug. 8, 2022, 5 p.m.
Part 2 of the code for the NeurIPS 2021 paper "Learning Generalized Gumbel-max Causal Mechanisms"
created at Nov. 12, 2021, 6:16 p.m.
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
created at Dec. 21, 2022, 9:20 a.m.
LoRA for arbitrary JAX models and functions
created at April 21, 2023, 11:50 a.m.
A selection of neural network models ported from torchvision for JAX & Flax.
created at June 14, 2020, 4:34 p.m.
A Python package of computer vision models for the Equinox ecosystem.
created at July 24, 2022, 10:02 p.m.
Add a tqdm progress bar to your JAX scans and loops.
created at Jan. 16, 2023, 4:54 p.m.
Composable kernels for scikit-learn implemented in JAX.
created at March 10, 2020, 6:59 p.m.
Unofficial JAX implementations of deep learning research papers
created at Jan. 8, 2022, 1:40 p.m.
Probabilistic Programming and Nested sampling in JAX
created at July 31, 2020, 6 p.m.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
created at May 16, 2021, 1:34 p.m.