Flax is a neural network library for JAX that is designed for flexibility.
-
Updated
May 9, 2024 - Python
Flax is a neural network library for JAX that is designed for flexibility.
A lightweight real-time model for South Korean license plate recognition.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Coordinates in JAX
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Orbax provides common utility libraries for JAX users.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
Distributed ML Training and Fine-Tuning on Kubernetes
Auto-differentiable and hardware-accelerated force density method
Normalizing-flow enhanced sampling package for probabilistic inference in Jax
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
Pixel implementation for the non-parametric component separation
Build, fit, and sample from cognitive models with JAX + NumPyro.
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."