flax topic

List flax repositories

pyprobml

6.3k
Stars
1.5k
Forks
Watchers

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

get-started-with-JAX

578
Stars
92
Forks
Watchers

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 t...

flaxOptimizers

29
Stars
1
Forks
Watchers

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

transformers

129.0k
Stars
25.6k
Forks
1.1k
Watchers

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

uvadlc_notebooks

2.2k
Stars
525
Forks
Watchers

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

NanoRange

2.2k
Stars
525
Forks
Watchers

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

flaxmodels

224
Stars
24
Forks
Watchers

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

score_sde

1.3k
Stars
186
Forks
Watchers

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

FlaxEngine

5.5k
Stars
539
Forks
Watchers

Flax Engine – multi-platform 3D game engine

jaxrl

590
Stars
62
Forks
Watchers

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

FlaxAPI

135
Stars
27
Forks
Watchers

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