#计算机科学#🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
Collection of example projects for Flax Engine
#编辑器#Old repository with C# Editor and C# API for creating games in Flax Engine
#计算机科学#Gaussian processes in JAX and Flax.
#计算机科学#Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
general purpose programming language, in the vein of C++
GPT implementation in Flax
Implementing the Denoising Diffusion Probabilistic Model in Flax
#计算机科学#JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
#计算机科学#Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Pretraining Roberta on Marathi using TPU and JAX, Flax and Optax
#自然语言处理#Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Tutorial to pretrain & fine-tune a 🤗 Flax T5 model on a TPUv3-8 with GCP