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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 5 years ago
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- Unofficial JAX implementations of deep learning research papers☆161Jun 25, 2022Updated 3 years ago
- This project compares the performance of Swin-Transformer v2 implemented in JAX and PyTorch.☆12Jun 8, 2022Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Jax/Flax implementation of DeiT and DeiT-III (ViT)☆19Dec 21, 2024Updated last year
- Shaping capabilities with token-level pretraining data filtering☆85Jan 28, 2026Updated last month
- execute shell commands in the Unity Editor☆11May 12, 2025Updated 10 months ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Neural Time Series Analysis☆14Nov 21, 2022Updated 3 years ago
- A standard bare-bone ROS Gazebo simulator for the Franka Emika Panda robot built using inbuilt Gazebo ROS controllers and RobotHW interfa…☆11May 3, 2021Updated 4 years ago
- ☆367Apr 12, 2024Updated last year
- Jax/Flax implementation of Denoising Diffusion Implicit Models☆20Jul 18, 2022Updated 3 years ago
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Procgen2: A community maintained fork of procgen☆12Aug 25, 2022Updated 3 years ago
- ☆10Sep 21, 2024Updated last year
- A Pytorch Lightning WGAN-gp to generate faces☆11Jan 26, 2021Updated 5 years ago
- ☆14Mar 31, 2025Updated 11 months ago
- Gradient Informed, GPU Accelerated Lens modelling (GIGALens) -- a package for fast Bayesian inference on strong gravitational lenses.☆12Oct 7, 2025Updated 5 months ago
- Stream smartphone sensor data with FastAPI, Kafka, ksqlDB, and Docker.☆11Aug 3, 2023Updated 2 years ago
- Code for "Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation" @ NeurIPS 2023☆25Oct 12, 2023Updated 2 years ago
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆13Jun 24, 2023Updated 2 years ago
- 🕸 GlotCC Dataset and Pipline -- NeurIPS 2024☆20Apr 6, 2025Updated 11 months ago
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Complex-Edit: CoT-Like Instruction Generation for Complexity-Controllable Image Editing Benchmark☆28Apr 22, 2025Updated 11 months ago
- Paper introducing jax-cosmo☆13Apr 27, 2023Updated 2 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…☆41Mar 7, 2026Updated 2 weeks ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 4 months ago
- Official repository of the "Attention-Propagation Network for Egocentric Heatmap to 3D Pose Lifting" (CVPR 2024 Highlight)☆14Dec 24, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Dec 30, 2022Updated 3 years ago
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 3 years ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated 9 months ago
- Concept Learning Dynamics☆16Oct 29, 2024Updated last year
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 5 months ago
- dolfinx_materials is a Python add-on package to the dolfinx interface to the FEniCSx project. It enables the user to define complex mater…☆27Feb 17, 2026Updated last month
- ☆15Oct 29, 2019Updated 6 years ago