8bitmp3
google/flax/docs 🤖 jax-ml/jax/docs 🤖 google/gemma/docs 🤖 tensorflow/docs
tensorflow, jax, flax
8bitmp3's Stars
probml/shifty
Distribution Shift
google-research/perch
metalwhale/FlaxHolidays
A collection of my experiences with Flax.
google/brain-tokyo-workshop
🧠🗼
RobertTLange/evosax
Evolution Strategies in JAX 🦎
conceptofmind/DeepViT-flax
Implementation of Deep Vision Transformer in Flax
google/neural-tangents
Fast and Easy Infinite Neural Networks in Python
probml/dynamax
State Space Models library in JAX
blackjax-devs/blackjax
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
probml/pml-book
"Probabilistic Machine Learning" - a book series by Kevin Murphy
araffin/sbx
SBX: Stable Baselines Jax (SB3 + Jax)
teticio/audio-diffusion
Apply diffusion models using the new Hugging Face diffusers package to synthesize music instead of images.
fchollet/stable-diffusion-tensorflow
TensorFlow/Keras port of Stable Diffusion
openai/whisper
Robust Speech Recognition via Large-Scale Weak Supervision
karpathy/nn-zero-to-hero
Neural Networks: Zero to Hero
felixgontier/dcase-2022-baseline
Talendar/flappy-bird-gym
An OpenAI Gym environment for the Flappy Bird game
ASEM000/pytreeclass
Visualize, create, and operate on pytrees in the most intuitive way possible.
AakashKumarNain/diffusion_models
All about the fundamentals and working of Diffusion Models
e3nn/e3nn-jax
jax library for E3 Equivariant Neural Networks
instadeepai/jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
google-deepmind/dm_control
Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
cgarciae/jax_metrics
A metrics library for the JAX ecosystem
paganpasta/eqxvision
A Python package of computer vision models for the Equinox ecosystem.
8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
carbon-language/carbon-lang
Carbon Language's main repository: documents, design, implementation, and related tools. (NOTE: Carbon Language is experimental; see README)
vopani/jaxton
100 exercises to learn JAX
magenta/music-spectrogram-diffusion
AtsushiSakai/PythonRobotics
Python sample codes for robotics algorithms.
ange-yaghi/engine-sim
Combustion engine simulator that generates realistic audio.