21 Open source projects that are alternatives of or similar to flaxOptimizers

Collection of example projects for Flax Engine

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.

Toy implementations of some popular ML optimizers using Python/JAX

Neutron: A pytorch based implementation of Transformer and its variants.

KoCLIP: Korean port of OpenAI CLIP, in Flax

Chef Transformer 🍲 .

Summarize Massive Datasets using Submodular Optimization

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

Improved Hypergradient optimizers, providing better generalization and faster convergence.

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

Nx-powered Neural Networks

Unofficial JAX implementations of deep learning research papers

JAX implementations of core Deep RL algorithms

Lookahead mechanism for optimizers in Keras.

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

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

Flax Engine – multi-platform 3D game engine

Python code for "Machine learning: a probabilistic perspective" (2nd edition)

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)

🧑‍🏫 50! Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠