Implementation of numerous Vision Transformers in Google's JAX and Flax.
☆22Aug 30, 2022Updated 3 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:
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- A combination of RoBERTa trained from scratch on masking histone modification patterns rather than the English language and XGBoost, pred…☆13Apr 22, 2021Updated 4 years ago
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 5 years ago
- Jax/Flax implementation of DeiT and DeiT-III (ViT)☆19Dec 21, 2024Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Mar 21, 2025Updated 11 months ago
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- ☆12Oct 7, 2020Updated 5 years ago
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆41May 6, 2025Updated 9 months ago
- Shaping capabilities with token-level pretraining data filtering☆83Jan 28, 2026Updated last month
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Jan 4, 2024Updated 2 years ago
- Neural Error Mitigation of Near-Term Quantum Simulations (arXiv:2105.08086)☆10Jul 6, 2022Updated 3 years ago
- Here, I provided the solution for exercises of IBM Quantum Challenge 2020☆10Oct 27, 2020Updated 5 years ago
- Material associated with Physics Report "Data science applications to string theory"☆11Jun 20, 2023Updated 2 years ago
- ☆91Sep 19, 2022Updated 3 years ago
- ☆367Apr 12, 2024Updated last year
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- 기획자와 마케터를 위한 이벤트 댓글 분석 - feat. 인프런 새해 다짐 이벤트☆11Apr 22, 2020Updated 5 years ago
- Efficient implementation of a parallel bucket-sort with OpenMP☆10Mar 19, 2017Updated 8 years ago
- SAR Backprojection Implementations for the GOTCHA data set☆12Jan 28, 2024Updated 2 years ago
- Gradient Informed, GPU Accelerated Lens modelling (GIGALens) -- a package for fast Bayesian inference on strong gravitational lenses.☆12Oct 7, 2025Updated 4 months ago
- angle-sequence☆12Apr 3, 2020Updated 5 years ago
- This project compares the performance of Swin-Transformer v2 implemented in JAX and PyTorch.☆12Jun 8, 2022Updated 3 years ago
- ☆10Jul 22, 2024Updated last year
- ☆13Updated this week
- execute shell commands in the Unity Editor☆11May 12, 2025Updated 9 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Jun 20, 2021Updated 4 years ago
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Jan 19, 2025Updated last year
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆54Feb 12, 2026Updated 3 weeks ago
- Reference list of email processing resources; focus on preservation and PII handling☆14Apr 20, 2022Updated 3 years ago
- ROS Mobile Inverted Pendulum☆11Apr 1, 2019Updated 6 years ago
- ☆12Jan 12, 2015Updated 11 years ago
- [3DV 2026] GIGA: Generalizable Sparse Image-driven Gaussian Humans☆18Jan 28, 2026Updated last month
- Official implementation of the paper “Endowing Vision-Language Models with System 2 Thinking for Fine-Grained Visual Recognition,” AAAI 2…☆32Jan 30, 2026Updated last month
- >>PhysWikiQuiz<< - a Physics Question Generation and Interrogation System☆11Feb 25, 2023Updated 3 years ago
- ☆10Mar 10, 2023Updated 2 years ago
- REOBench: Benchmarking Robustness of Earth Observation Foundation Models☆25Oct 28, 2025Updated 4 months ago
- Flash cards chrome extensions☆13Jul 12, 2022Updated 3 years ago