vballoli / vit-flaxView external linksLinks
Implementation of Vision Transformers in Flax
☆18Oct 12, 2020Updated 5 years ago
Alternatives and similar repositories for vit-flax
Users that are interested in vit-flax are comparing it to the libraries listed below
Sorting:
- 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
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Contact-Aware Symplectic Integrator Network☆16Mar 22, 2023Updated 2 years ago
- Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.☆20Aug 4, 2021Updated 4 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Attention mechanism with MNIST dataset☆17Jun 11, 2021Updated 4 years ago
- Config files for my GitHub profile.☆24Mar 23, 2025Updated 10 months ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Jan 4, 2024Updated 2 years ago
- ☆28Jan 12, 2022Updated 4 years ago
- Multi-agent Reinforcement Learning, 14th in 701 teams - NeurIPS 2024 Competition☆14Mar 13, 2025Updated 11 months ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 6 months ago
- Directed masked autoencoders☆14Feb 5, 2026Updated last week
- Image augmentation library for Jax☆41Apr 9, 2024Updated last year
- Source code and datasets for Circuit Design Completion using GNNs paper☆10Jan 26, 2023Updated 3 years ago
- Learning Generalized Physical Representation from a Few Examples☆16Updated this week
- ☆11Aug 4, 2019Updated 6 years ago
- Water wave models in one dimension☆10Jan 4, 2026Updated last month
- This project shows how to build a simple handwriting recognizer in Keras with the IAM dataset.☆13Aug 15, 2021Updated 4 years ago
- Iterated Local Search (ILS) metaheuristic applied to the Prize Collecting Traveling Salesman Problem (PCTSP).☆10Jul 15, 2019Updated 6 years ago
- Gym wrapper for pysc2☆10Sep 16, 2022Updated 3 years ago
- Code for the paper "SizeShiftReg: a Regularization Method for Improving Size-Generalization in Graph Neural Networks"☆12Jan 17, 2023Updated 3 years ago
- A curated list of papers & resources linked to concept learning☆12Aug 9, 2023Updated 2 years ago
- Virtual notebook that Evan uses for his PhD thesis.☆11Sep 5, 2025Updated 5 months ago
- Official implementation of the paper “Endowing Vision-Language Models with System 2 Thinking for Fine-Grained Visual Recognition,” AAAI 2…☆32Jan 30, 2026Updated 2 weeks ago
- Reimplementation of VFDT and EFDT. Our codebase is simple, concise, easy to reproduce with strong performances.☆10Feb 1, 2021Updated 5 years ago
- an optimizing curry compiler☆14Nov 27, 2022Updated 3 years ago
- GDAL + OpenCV + CUDA☆10Nov 10, 2017Updated 8 years ago
- Arete is an experimental programming language.☆12Oct 6, 2023Updated 2 years ago
- Summer Scheming!!!!!!☆11Aug 20, 2020Updated 5 years ago
- BIOBOT: A Fall Detection System (FDS) using Artificial Intelligence☆13Jan 19, 2019Updated 7 years ago
- ☆12Nov 29, 2021Updated 4 years ago
- Frame-independent vector-cloud neural network for nonlocal constitutive modelling on arbitrary grids.☆11Oct 31, 2021Updated 4 years ago
- tensorflow implementation of Neural Oblivious Decision Ensembles☆10Jan 8, 2021Updated 5 years ago
- Supporting code for the paper «Leveraging molecular structure and bioactivity with chemical language models for drug design»☆12Feb 22, 2022Updated 3 years ago
- C++ implementation of an incremental primal-dual assignment problem solver.☆12Dec 13, 2024Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆104May 17, 2022Updated 3 years ago
- Implementation of Physics-Informed PointNet (PIPN) for weakly-supervised learning of 2D linear elasticity (plane stress) on multiple sets…☆12Apr 11, 2024Updated last year
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago