Implementation of Vision Transformers in Flax
☆18Oct 12, 2020Updated 6 years ago
Alternatives and similar repositories for vit-flax
Users that are interested in vit-flax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 5 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆23Dec 22, 2020Updated 5 years ago
- Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.☆20Aug 4, 2021Updated 5 years ago
- Contact-Aware Symplectic Integrator Network☆19Mar 22, 2023Updated 3 years ago
- Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API☆13Sep 25, 2021Updated 5 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆46Jul 19, 2025Updated last year
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Config files for my GitHub profile.☆25Mar 23, 2025Updated last year
- JAX implementation of Large Language Models. You can train GPT-2-like model with 青空文庫 (aozora bunko-clean dataset) or any other text dat…☆13Aug 5, 2024Updated 2 years ago
- JRDB dataset: trajectory prediction baselines and data preprocessing☆13Nov 7, 2023Updated 2 years ago
- A scaffold to speed up launching a flask project.☆15Jun 3, 2026Updated 4 months ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Image augmentation library for Jax☆44Apr 9, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Reinforcement Learning Environments for Omniverse Isaac Gym☆10May 9, 2023Updated 3 years ago
- Iterated Local Search (ILS) metaheuristic applied to the Prize Collecting Traveling Salesman Problem (PCTSP).☆10Jul 15, 2019Updated 7 years ago
- 2D gridmap tools☆18Nov 9, 2021Updated 4 years ago
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago
- Multi-agent Reinforcement Learning, 14th in 701 teams - NeurIPS 2024 Competition☆15Mar 13, 2025Updated last year
- Attention mechanism with MNIST dataset☆17Jun 11, 2021Updated 5 years ago
- ☆11Apr 16, 2020Updated 6 years ago
- ResNet-RS models rewritten in Tensorflow / Keras functional API.☆27Jul 26, 2023Updated 3 years ago
- Augmenting dynamical systems by combining them with deep reinforcement learning in order to improve their final performance.☆11Jun 27, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- AlphaGo inspired TSP Heuristic Solver☆14Feb 5, 2020Updated 6 years ago
- Jax implementation of VIT-VQGAN☆10Jan 25, 2024Updated 2 years ago
- Using Deep Reinforcement Learning Project Repository☆10Nov 21, 2022Updated 3 years ago
- Ball & beam OpenAI gym environments☆15Mar 4, 2020Updated 6 years ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆26Jun 27, 2023Updated 3 years ago
- ☆21Feb 22, 2025Updated last year
- Dataset generation scripts for “Path Planning using Neural A* Search” presented in ICML-21☆27Jan 30, 2023Updated 3 years ago
- Code for our paper on doing resource allocation with graph neural networks☆30Jun 24, 2021Updated 5 years ago
- Router Network Operating System☆18Oct 14, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆58Jan 20, 2023Updated 3 years ago
- Object oriented Tensorflow implementation of the famous LeNet5 network by Yann Lecun.☆16Mar 22, 2018Updated 8 years ago
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆34Sep 15, 2023Updated 3 years ago
- A Wadler–Lindig pretty printer for Python☆48Apr 20, 2026Updated 5 months ago
- Simple HTTP golang framework☆13Jul 4, 2019Updated 7 years ago
- Train very large language models in Jax.☆209Oct 21, 2023Updated 2 years ago