flax

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

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

Flax Engine – multi-platform 3D game engine

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

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Long Range Arena for Benchmarking Efficient Transformers

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

Original Implementation of Prompt Tuning from Lester, et al, 2021

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

Tevatron - A flexible toolkit for dense retrieval research and development.

general purpose programming language, in the vein of C++

Old repository with C# Editor and C# API for creating games in Flax Engine

CLU lets you write beautiful training loops in JAX.

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

Unofficial JAX implementations of deep learning research papers

KoCLIP: Korean port of OpenAI CLIP, in Flax

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)