Stars
2
Forks
1
Language
Python
Last Updated
Aug 10, 2022
Similar Repos
Repo | Language | Stars | Description | Updated At |
---|---|---|---|---|
Python | 38 | Jax/Flax implementation of Variational-DiffWave. | Apr 16, 2023 | |
Jupyter Notebook | 9 | Various reinforcement learning algorithms written in Jax + Flax | May 09, 2023 | |
Python | 54 | Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation | Apr 24, 2023 | |
Python | 2 | Jax/Flax implementation practice for various neural networks | Mar 22, 2023 | |
Python | 3 | Implementation of the SOTA Transformer architecture from PaLM - Scaling Language Modeling with Pathways in … | Jul 20, 2022 | |
Jupyter Notebook | 25 | Implementation of several Generative Adversarial Networks in JAX / Flax | Apr 23, 2023 | |
Python | 10 | FID computation in Jax/Flax. | Jul 31, 2022 | |
None | 2 | FID computation in Jax/Flax. | Jul 27, 2022 | |
Python | 3 | A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training | Oct 27, 2023 | |
Python | 2 | Implementation of numerous Vision Transformers in Google's JAX and Flax. | Aug 05, 2022 | |
Python | 15 | Jax/Flax rewrite of Karpathy's nanoGPT | Mar 21, 2023 | |
Python | 16 | Local Attention - Flax module for Jax | Jun 16, 2022 | |
Python | 6 | CIFAR10 ResNets implemented in JAX+Flax | May 16, 2023 | |
Python | 3 | Recipe Generation Model - Flax/JAX Projects | Sep 22, 2022 | |
Python | 3 | Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax. | Aug 30, 2023 | |
Python | 2 | Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions. … | Jan 26, 2023 | |
Jupyter Notebook | 2 | Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax. | Feb 08, 2024 | |
Python | 3 | 🔐 Serialize JAX/Flax models with `safetensors` | Dec 24, 2022 | |
Python | 4 | Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...) | Mar 26, 2023 | |
Python | 2 | Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku | Jun 22, 2022 | |
Jupyter Notebook | 369 | JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces. | Aug 26, 2022 | |
Python | 37 | Pre-training script for BART in JAX/Flax | Aug 11, 2022 | |
Python | 3 | Create nice dash/zeal docset for JAX+Flax | Oct 11, 2022 | |
Jupyter Notebook | 4 | Colab Notebooks for JAX/Flax/Optax ML Book | Feb 28, 2023 | |
Python | 12 | Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation | Jan 08, 2023 | |
Python | 25 | Shared code for training sentence embeddings with Flax / JAX | Feb 14, 2023 | |
Python | 150 | Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc. | Aug 19, 2022 | |
Python | 13 | GPT implementation in Flax | Sep 02, 2022 | |
Jupyter Notebook | 15 | JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et … | Apr 14, 2023 | |
Python | 66 | Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control … | Sep 01, 2022 | |
Python | 10 | Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging … | Nov 17, 2022 | |
Python | 22 | A CLIP conditioned Decision Transformer. | Aug 05, 2022 | |
Python | 8 | comparing Jax (haiku and flax) to PyTorch on simple DL tasks. | Mar 25, 2023 | |
Python | 6 | Minimal PyTorch implementation of Point Transformer from "Point Transformer" by Zhao et al. | Feb 02, 2022 | |
Rust | 3 | Pure Rust implementation of a minimal Generative Pretrained Transformer | May 30, 2023 | |
Python | 151 | A pure-functional implementation of a machine learning transformer model in Python/JAX | Jul 24, 2022 | |
None | 16 | Transformer in RL for decision-making | May 20, 2023 | |
Python | 29 | Unofficial code for online decision transformer | May 07, 2023 | |
Python | 114 | EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax | Aug 07, 2022 | |
Python | 28 | Repository for Multilingual-VQA task created during HuggingFace JAX/Flax community week. | Apr 04, 2023 | |
Python | 3584 | Flax is a neural network library for JAX that is designed for flexibility. | Oct 06, 2022 | |
None | 2 | Repository for fine-tuning Transformers 🤗 based seq2seq speech models in JAX/Flax. | Jan 18, 2023 | |
Jupyter Notebook | 15 | Repository for fine-tuning Transformers 🤗 based seq2seq speech models in JAX/Flax. | Jun 05, 2022 | |
None | 5 | Flax is a neural network library for JAX that is designed for flexibility. | Apr 19, 2023 | |
Python | 2 | Applying regular transformer and decision transformer on solving the Rubik's cube. A paper is also … | May 09, 2023 | |
Python | 3 | JAX implementation of BART, aiming to demonstrate how Transformer-based models can be implemented using JAX … | Feb 09, 2023 | |
Jupyter Notebook | 6 | Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax. | Jan 21, 2023 | |
Jupyter Notebook | 2 | Practice repository for jax and flax implementing NeRF: Representing Scenes as Neural Radiance Fields | Mar 15, 2022 | |
Python | 15 | Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX. | Mar 21, 2023 | |
C# | 2 | A little B-Spline implementation for Flax | Aug 20, 2023 |