Popular GitHub Repositories for Jax Relax

? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Deep Learning for humans

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

? Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

? A ranked list of awesome machine learning Python libraries. Updated weekly.

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Trax — Deep Learning with Clear Code and Speed

Python code for "Probabilistic Machine learning" book by Kevin Murphy

Flax is a neural network library for JAX that is designed for flexibility.

Relax! Flux is the ML library that doesn't make you tensor

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

Scenic: A Jax Library for Computer Vision Research and Beyond

A library for scientific machine learning and physics-informed learning

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Training and serving large-scale neural networks with auto parallelization.

JAX-based neural network library

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX