☆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 5 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
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- EncryptedClipboard☆13Sep 24, 2020Updated 5 years ago
- ☆35Sep 6, 2025Updated last year
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆174Nov 11, 2025Updated 10 months ago
- Code for the reproduction of counting manifolds☆16Feb 26, 2026Updated 6 months ago
- Particle filtering in JAX☆17Aug 27, 2026Updated 2 weeks ago
- List of new Project Fundraising Opportunities for NumFOCUS Sponsored Projects☆14May 14, 2026Updated 3 months ago
- GGUF parser in Python☆28May 1, 2026Updated 4 months ago
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Aug 24, 2026Updated 2 weeks ago
- Pytorch-like dataloaders for JAX.☆105Dec 16, 2025Updated 8 months 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.
- 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 2 years ago
- Sparse Autoencoder Training Library☆58May 1, 2025Updated last year
- Next Generation Experimental Tracking for Machine Learning Operations☆17Aug 28, 2026Updated 2 weeks ago
- Dockerfile and instructions for human pose estimation implementation using Caffe, OpenCV 3.1.0 and Python 2.7.☆12Mar 3, 2019Updated 7 years ago
- a Jax quantization library☆133Updated this week
- Tokamax: A GPU and TPU kernel library.☆285Updated this week
- [COLM 2025: 1st Workshop on the Application of LLM Explainability to Reasoning and Planning] Latent Chain-of-Thought? Decoding the Depth-…☆21Aug 19, 2026Updated 3 weeks ago
- ☆19Dec 4, 2025Updated 9 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆34Jun 11, 2025Updated last year
- Improving Neural Text Generation with Reinforcement Learning☆23Jan 13, 2021Updated 5 years ago
- Minimal yet performant LLM examples in pure JAX☆277Aug 23, 2026Updated 2 weeks ago
- A Snakemake workflow for Fermi-LAT data reduction☆12Aug 12, 2025Updated last year
- LaTeX course in RANEPA☆13Nov 7, 2020Updated 5 years ago
- 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
- All the tools that allow me to never ever open up Final Cut☆11Feb 16, 2025Updated last year
- 北京大学跑步软件PKURUNNER破解☆21May 13, 2021Updated 5 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.
- nyc is so back☆21Jun 27, 2025Updated last year
- a common lisp library for data analysis and manipulation☆12May 28, 2023Updated 3 years ago
- Convert StableHLO models into Apple Core ML format☆22Updated this week
- Common Lisp implementation of the Erlang External Term Format☆16Dec 31, 2022Updated 3 years ago
- A library of speech gadgets.☆16Oct 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"☆85Dec 18, 2025Updated 8 months ago