dogear

enter for all results · esc to close

Accurate Quantized Training

github.comsite

Tools and libraries for running and analyzing neural network quantization experiments in JAX and Flax.

visit github.com ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

similar

  1. FLAX github.com
    tool category #python-data-science link ↗

    A neural network library for JAX that is designed for flexibility.

  2. flaxmodels github.com
    tool category #jax link ↗

    Pretrained models for Jax/Flax.

  3. jax-resnet github.com
    tool category #jax link ↗

    Implementations and checkpoints for ResNet variants in Flax.

  4. Deep Learning tutorials with JAX+Flax by Phillip Lippe uvadlc-notebooks.readthedocs.io
    site category #jax link ↗

    A series of notebooks explaining various deep learning concepts, from basics (e.g. intro to JAX/Flax, activiation functions) to recent advances (e.g., Vision Transformers, SimCLR), with translations to PyTorch.

  5. FedJAX github.com
    tool category #jax link ↗

    Federated learning in JAX, built on Optax and Haiku.

  6. Tutorial: image classification with JAX and Flax Linen by 8bitmp3 github.com
    tool category #jax link ↗

    Learn how to create a simple convolutional network with the Linen API by Flax and train it to recognize handwritten digits.

JAX › Models and Projects > JAX: “Tools and libraries for running and analyzing neural network quantization experiments in JAX and Flax.”