All Categories → No Category → flax

Top 15 flax open source projects

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

Flax Engine – multi-platform 3D game engine

✭ 3,127

C++C#ASP.NET GLSL c HLSL game-development game-engine gamedev engine multi-platform flax flax-engine flaxengine

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

✭ 4,197

python Jupyter Notebook machine-learning tensorflow pytorch colab pml flax jupyter-notebooks pymc3 pyro jax numpyro

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

✭ 61

python machine-learning computer-vision deep-learning neural-networks resnet flax jax

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

✭ 49

python flax normalizing-flows jax diffusion-models score-matching neural-odes score-based-generative-modeling neurips-2021 score-based-generative-models

A collection of optimizers, some arcane others well known, for Flax.

✭ 21

python flax optimizers

Collection of example projects for Flax Engine

✭ 50

C#smalltalk gamedev visual-studio templates samples flax flax-engine

get-started-with-JAX

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.

✭ 229

Jupyter Notebook machine-learning tutorial deep-learning jupyter numpy flax haiku lax jax xla optax learn-jax

KoCLIP: Korean port of OpenAI CLIP, in Flax

✭ 80

Jupyter Notebook python flax jax roberta vision-transformer openai-clip

chef-transformer

Chef Transformer 🍲 .

✭ 29

python Jupyter Notebook CSS recipes transformers flax recipe-generator jax huggingface t5 recipe-generation google-t5

efficientnet-jax

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

✭ 114

python Dockerfile flax mixnet tpu jax mobilenetv2 mobilenetv3 efficientnet objax flax-linen

uvadlc notebooks

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

✭ 901

Jupyter Notebook tutorial deep-learning tutorials pytorch flax jax pytorch-lightning optax uvadlc

Unofficial JAX implementations of deep learning research papers

✭ 108

python machine-learning computer-vision deep-learning transformers artificial-intelligence convolutional-neural-networks flax jax research-paper-implementation

JAX implementations of core Deep RL algorithms

✭ 61

python Makefile reinforcement-learning deep-learning deep-reinforcement-learning flax deepmind mpo sac actor-critic maximum-a-posteriori-estimation mujoco jax td3 soft-actor-critic

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

✭ 43

Jupyter Notebook python machine-learning reinforcement-learning deep-learning automatic-differentiation deep-reinforcement-learning dqn gym flax haiku sac model-based-rl jax model-based-reinforcement-learning soft-actor-critic dm-haiku

1-15 of 15 flax projects