JAX-Flax-Tutorial-Image-Classification-with-Linen
How to use the Flax Linen API to build a convolutional neural network model and
Jupyter Notebook24
last year
jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with con
Jupyter Notebook631mit
2 years ago
batch-reinforcement-learningbehavioral-cloningcontinuous-control
flaxvision
A selection of neural network models ported from torchvision for JAX & Flax.
Python44apache-2.0
4 years ago
safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Python42mit
6 months ago
flaxflax-serializationhaiku
flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Python238
last year
deep-learningflaxgpt2
GNN-for-Combinatorial-Optimization
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired G
Jupyter Notebook42mit
2 years ago
combinatorial-optimizationdeep-learningflax
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-tr
Python2409apache-2.0
3 months ago
chatbotdeep-learningflax
get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and H
Jupyter Notebook662mit
12 months ago
deep-learningflaxhaiku
jax-resnet
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and
Python104mit
2 years ago
computer-visiondeep-learningflax
JAX-Flax-Tutorial-Image-Classification-with-Linen
How to use the Flax Linen API to build a convolutional neural network model and
Jupyter Notebook24
last year
jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with con
Jupyter Notebook631mit
2 years ago
batch-reinforcement-learningbehavioral-cloningcontinuous-control
flaxvision
A selection of neural network models ported from torchvision for JAX & Flax.
Python44apache-2.0
4 years ago
safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Python42mit
6 months ago
flaxflax-serializationhaiku
flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Python238
last year
deep-learningflaxgpt2
GNN-for-Combinatorial-Optimization
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired G
Jupyter Notebook42mit
2 years ago
combinatorial-optimizationdeep-learningflax
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-tr
Python2409apache-2.0
3 months ago
chatbotdeep-learningflax
get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and H
Jupyter Notebook662mit
12 months ago
deep-learningflaxhaiku
jax-resnet
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and
Python104mit
2 years ago
computer-visiondeep-learningflax