☆30Oct 26, 2024Updated last year
Alternatives and similar repositories for flaxattention
Users that are interested in flaxattention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆34Jun 3, 2024Updated 2 years ago
- Translation of the databricks-dolly-15k dataset to Chinese for commercial use.☆19Apr 17, 2023Updated 3 years ago
- ☆13Aug 7, 2021Updated 4 years ago
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 3 years ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Jan 31, 2023Updated 3 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.
- EncryptedClipboard☆13Sep 24, 2020Updated 5 years ago
- ☆35Sep 6, 2025Updated 10 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆174Nov 11, 2025Updated 8 months ago
- Code for the reproduction of counting manifolds☆16Feb 26, 2026Updated 5 months ago
- JAX Implementations of Descript Audio Codec and EnCodec☆38Mar 30, 2025Updated last year
- Particle filtering in JAX☆17Jun 12, 2026Updated last month
- Final repository for the project "Unsupervised Machine Learning for the Classification of Astrophysical X-ray Sources". V. S. Pérez-Díaz,…☆10Jan 23, 2024Updated 2 years ago
- 3D Gaussian Splatting in JAX☆76May 30, 2024Updated 2 years ago
- Microbenchmarking hyperparameter tuning for JAX functions.☆22Jul 7, 2026Updated 3 weeks ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- 2021 Spring☆18Oct 12, 2024Updated last year
- Computer Science, Data Science and ML Fundamentals☆11May 30, 2025Updated last year
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Feb 27, 2023Updated 3 years ago
- triton ver of gqa flash attn, based on the tutorial☆12Aug 4, 2024Updated last year
- Sparse Autoencoder Training Library☆58May 1, 2025Updated last year
- Next Generation Experimental Tracking for Machine Learning Operations☆17Updated this week
- Tokamax: A GPU and TPU kernel library.☆255Updated this week
- a Jax quantization library☆127Updated this week
- ☆19Dec 4, 2025Updated 7 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆34Jun 11, 2025Updated last year
- Minimal yet performant LLM examples in pure JAX☆269Jul 4, 2026Updated 3 weeks ago
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆369Updated this week
- A Snakemake workflow for Fermi-LAT data reduction☆12Aug 12, 2025Updated 11 months ago
- VTSCat is the catalogue of all published observations of the VERITAS collaboration.☆12Jun 16, 2026Updated last month
- Official Implementation for the ICLR2023 paper "Fuzzy Alignments in Directed Acyclic Graph for Non-autoregressive Machine Translation"☆14Mar 1, 2023Updated 3 years ago
- MXNET + OpenAI Gym implementation of A3C from "Asynchronous Methods for Deep Reinforcement Learning"☆11Apr 10, 2017Updated 9 years ago
- A simple log aggregation tool☆16Aug 30, 2025Updated 10 months ago
- Jump ReLU☆12Apr 8, 2019Updated 7 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.
- Math is like the secret sauce behind Machine Learning and AI. It helps us make sense of data, find patterns, and make predictions. So, if…☆15Sep 12, 2023Updated 2 years ago
- a common lisp library for data analysis and manipulation☆12May 28, 2023Updated 3 years ago
- Convert StableHLO models into Apple Core ML format☆22Jun 28, 2026Updated last month
- Common Lisp implementation of the Erlang External Term Format☆16Dec 31, 2022Updated 3 years ago
- A library of speech gadgets.☆15Oct 15, 2022Updated 3 years ago
- a simple tool to translate caffe model to keras model☆10Oct 26, 2015Updated 10 years ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆86Dec 18, 2025Updated 7 months ago