Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`

Python42mit

6 months ago

flaxflax-serializationhaiku

FID computation in Jax/Flax.

Python24apache-2.0

4 months ago

fidflaxfrechet-inception-distance

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-tr

Python2409apache-2.0

3 months ago

chatbotdeep-learningflax

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

How to use the Flax Linen API to build a convolutional neural network model and

Jupyter Notebook24

last year

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with con

Jupyter Notebook631mit

2 years ago

batch-reinforcement-learningbehavioral-cloningcontinuous-control

A selection of neural network models ported from torchvision for JAX & Flax.

Python44apache-2.0

4 years ago

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

Python238

last year

deep-learningflaxgpt2

JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired G

Jupyter Notebook42mit

2 years ago

combinatorial-optimizationdeep-learningflax

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and

Python104mit

2 years ago

computer-visiondeep-learningflax