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:
- 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
- Includes additional materials for the following keras.io blog post.☆12Jun 23, 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
- Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.☆20Aug 4, 2021Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- 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
- 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
- Pytorch Lightning implementation of Neural Astar planner☆10Mar 27, 2022Updated 4 years ago
- Config files for my GitHub profile.☆25Mar 23, 2025Updated last year
- Hashing, searching, sorting, tree algorithms, dynamic programming, greedy algorithms, divide and conquer, random number generation, and r…☆10Apr 20, 2018Updated 8 years ago
- [ICRA 2020] Residual Reactive Navigation: Combining Classical and Learned Navigation Strategies for Deployment in Unknown Environments☆14Jan 23, 2020Updated 6 years ago
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆11Nov 23, 2023Updated 2 years ago
- 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.
- This project compares the performance of Swin-Transformer v2 implemented in JAX and PyTorch.☆12Jun 8, 2022Updated 4 years ago
- Google Calendar-style weekly view of your Claude Code sessions — local, live, read-only☆17Oct 10, 2026Updated 3 weeks ago
- Adds an ability to render LaTeX and AsciiMath formulas in SillyTavern chats.☆16Mar 17, 2026Updated 7 months ago
- JRDB dataset: trajectory prediction baselines and data preprocessing☆13Nov 7, 2023Updated 2 years ago
- Reinforcement Learning Environments for Omniverse Isaac Gym☆10May 9, 2023Updated 3 years ago
- ☆12Oct 19, 2017Updated 9 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆130Jan 4, 2024Updated 2 years ago
- Python implementation of Fast Marching Tree☆16Oct 25, 2020Updated 6 years ago
- 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.
- Image augmentation library for Jax☆44Apr 9, 2024Updated 2 years ago
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 4 years ago
- Fast Legged Trajectory Planner.☆12Jul 16, 2026Updated 3 months ago
- Attention mechanism with MNIST dataset☆17Jun 11, 2021Updated 5 years ago
- 2D gridmap tools☆18Nov 9, 2021Updated 4 years ago
- 本模型是基于Python3编写的工商信息图片识别文字模型。主要步骤包括去噪、分离水印、反色、二值化增强等，支持批量图片处理。☆13Mar 14, 2020Updated 6 years ago
- ☆11Apr 16, 2020Updated 6 years ago
- Generative models with JAX.☆22Feb 21, 2026Updated 8 months ago
- Learning Representations that Support Robust Transfer of Predictors☆20Nov 7, 2021Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)☆50Feb 15, 2022Updated 4 years ago
- Ball & beam OpenAI gym environments☆15Mar 4, 2020Updated 6 years ago
- Evaluation of LLMs as RPG Game Engines☆17May 15, 2025Updated last year
- Causal Analysis of Agent Behavior for AI Safety☆22Jun 27, 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
- ☆21Feb 22, 2025Updated last year
- Code for the paper: "Independent mechanism analysis, a new concept?"☆26Jun 27, 2023Updated 3 years ago