Stars
2
Forks
0
Language
Python
Last Updated
Mar 22, 2023
Similar Repos
Repo | Language | Stars | Description | Updated At |
---|---|---|---|---|
Jupyter Notebook | 25 | Implementation of several Generative Adversarial Networks in JAX / Flax | Apr 23, 2023 | |
Python | 2 | Variational Neural Networks JAX implementation | Jul 05, 2022 | |
Jupyter Notebook | 2 | Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax. | Feb 08, 2024 | |
Jupyter Notebook | 15 | JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et … | Apr 14, 2023 | |
Jupyter Notebook | 2 | Practice repository for jax and flax implementing NeRF: Representing Scenes as Neural Radiance Fields | Mar 15, 2022 | |
Jupyter Notebook | 3 | SIREN neural networks in Flax | Oct 27, 2021 | |
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 | Minimal Decision Transformer Implementation written in Jax (Flax). | Aug 10, 2022 | |
Python | 11 | Stainless neural networks in JAX | Nov 15, 2023 | |
Python | 16 | Local Attention - Flax module for Jax | Jun 16, 2022 | |
Python | 3584 | Flax is a neural network library for JAX that is designed for flexibility. | Oct 06, 2022 | |
None | 5 | Flax is a neural network library for JAX that is designed for flexibility. | Apr 19, 2023 | |
Python | 10 | FID computation in Jax/Flax. | Jul 31, 2022 | |
None | 2 | FID computation in Jax/Flax. | Jul 27, 2022 | |
Python | 2 | Implementation of numerous Vision Transformers in Google's JAX and Flax. | Aug 05, 2022 | |
Python | 2 | Variational Neural Networks Pytoch and JAX implementations | Jan 10, 2023 | |
None | 2 | Variational Neural Networks Pytoch and JAX implementations | May 19, 2023 | |
Python | 15 | Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX. | Mar 21, 2023 | |
Python | 15 | Jax/Flax rewrite of Karpathy's nanoGPT | Mar 21, 2023 | |
Python | 6 | CIFAR10 ResNets implemented in JAX+Flax | May 16, 2023 | |
Python | 3 | Recipe Generation Model - Flax/JAX Projects | Sep 22, 2022 | |
Python | 7 | An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN) | Jul 07, 2022 | |
Python | 3 | Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax. | Aug 30, 2023 | |
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 | |
Jupyter Notebook | 369 | JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces. | Aug 26, 2022 | |
Python | 11 | JAX implementation of Graph Attention Networks | Jul 13, 2022 | |
Python | 3 | 🔐 Serialize JAX/Flax models with `safetensors` | Dec 24, 2022 | |
Python | 4 | Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...) | Mar 26, 2023 | |
Python | 28 | Build neural networks for machine learning force fields with JAX | May 01, 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 | |
Jupyter Notebook | 5 | A short tutorial on Bayesian Neural Networks with JAX | Mar 03, 2023 | |
Python | 21 | A collection of graph neural networks implementations in JAX | May 09, 2023 | |
Python | 23 | A PyTorch implementation of various Graph Neural Networks (GNNs) for graph classification | Jun 29, 2022 | |
Python | 14 | An unofficial implementation of the Fourier Neural Operator in Flax | Jan 22, 2024 | |
Python | 150 | Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc. | Aug 19, 2022 | |
None | 2 | Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX. | Mar 12, 2023 | |
Python | 1098 | Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/ | Apr 27, 2023 | |
Python | 4 | Variational Neural Networks Pytorch implementation | Mar 11, 2023 | |
D | 2 | neural networks implementation in D | Dec 12, 2012 | |
Haskell | 2 | Haskell implementation of Neural Networks | Sep 26, 2022 | |
Python | 13 | GPT implementation in Flax | Sep 02, 2022 | |
Python | 3 | Implementation of the SOTA Transformer architecture from PaLM - Scaling Language Modeling with Pathways in … | Jul 20, 2022 | |
Python | 28 | Repository for Multilingual-VQA task created during HuggingFace JAX/Flax community week. | Apr 04, 2023 | |
JavaScript | 2 | artificial neural networks implementation using SGD | Jan 24, 2020 | |
Jupyter Notebook | 3 | Convolution Neural Networks implementation in TensorFlow | Feb 19, 2023 |