There are 1 repository under flax topic.
🤗 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
general purpose programming language, in the vein of C++
Old repository with C# Editor and C# API for creating games in Flax Engine
Tevatron - A flexible toolkit for dense retrieval research and development.
Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
KoCLIP: Korean port of OpenAI CLIP, in Flax
Unofficial JAX implementations of deep learning research papers
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)
JAX implementations of core Deep RL algorithms
Flax Engine Documentation
Collection of example projects for Flax Engine
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Jax/Flax implementation of Variational-DiffWave.
Chef Transformer 🍲 .
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
A collection of optimizers, some arcane others well known, for Flax.
RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week
Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
Fine-tuning a CLIP model on the ROCO medical dataset
Jax implementation of the AdaHessian optimizer
Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API
Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
Implementation of several Generative Adversarial Networks in JAX / Flax
Multiplayer sample with players lobby and chat in Flax Engine