☆30Oct 26, 2024Updated 2 years ago
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:
- ☆35Jun 3, 2024Updated 2 years ago
- Translation of the databricks-dolly-15k dataset to Chinese for commercial use.☆19Apr 17, 2023Updated 3 years ago
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 3 years ago
- EncryptedClipboard☆13Sep 24, 2020Updated 6 years ago
- Tokamax: A GPU and TPU kernel library.☆321Updated this week
- 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.
- Fast, Numerically Stable, and Auto-Differentiable Spectral Clipping via Newton-Schulz Iteration☆12Jun 21, 2025Updated last year
- SciPy 2025 tutorial on "Reproducible Machine Learning Workflows for Scientists with pixi"☆11Mar 3, 2026Updated 7 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆177Nov 11, 2025Updated 11 months ago
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- Code for the reproduction of counting manifolds☆16Feb 26, 2026Updated 8 months ago
- Minimal yet performant LLM examples in pure JAX☆286Aug 23, 2026Updated 2 months ago
- JAX Implementations of Descript Audio Codec and EnCodec☆38Aug 23, 2026Updated 2 months ago
- ☆11Feb 4, 2023Updated 3 years ago
- Repository with draft content for the Pydantic tutorial at SciPy 2023☆13Jul 25, 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.
- List of new Project Fundraising Opportunities for NumFOCUS Sponsored Projects☆14May 14, 2026Updated 5 months ago
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Oct 9, 2026Updated 3 weeks ago
- Pytorch-like dataloaders for JAX.☆105Dec 16, 2025Updated 10 months ago
- 3D Gaussian Splatting in JAX☆76May 30, 2024Updated 2 years ago
- Custom kernels in Triton language for accelerating LLMs☆27Apr 5, 2024Updated 2 years ago
- Next Generation Experimental Tracking for Machine Learning Operations☆18Oct 24, 2026Updated last week
- triton ver of gqa flash attn, based on the tutorial☆12Aug 4, 2024Updated 2 years ago
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆34Jun 11, 2025Updated last year
- Sparse Autoencoder Training Library☆59May 1, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A small (~1000LoC) embeddable LISP interpreter☆13Feb 28, 2025Updated last year
- ☆19Dec 4, 2025Updated 10 months ago
- ☆10Mar 12, 2018Updated 8 years ago
- a Jax quantization library☆140Updated this week
- An unnecessarily tiny and minimal implementation of GPT-2 in NumPy.☆11Feb 12, 2023Updated 3 years ago
- 使用 Vue CLI 重置的奥利奥！☆22Oct 5, 2021Updated 5 years ago
- VTSCat is the catalogue of all published observations of the VERITAS collaboration.☆12Jun 16, 2026Updated 4 months ago
- QWOP AI using Q-learning☆12Jul 13, 2016Updated 10 years ago
- Mxnet implementation of an ICLR 2018 paper: A new method of region embedding for text classification.☆10Oct 14, 2018Updated 8 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.
- Library for reading and processing ML training data.☆800Updated this week
- Common Lisp library that allows to run a child process and have its input, output and error avaliable as Gray streams.☆11Feb 10, 2017Updated 9 years ago
- Reversal Curse Experiment☆15Sep 24, 2023Updated 3 years ago
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆376Updated this week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆84Dec 18, 2025Updated 10 months ago
- Parse CL syntactic constructs in s-expression form (possibly represented as e.g. CSTs) and construct syntax trees☆10Nov 4, 2024Updated last year
- Question-Answering ranking with Deep Learning models (cDSSM, Convolutional, LSTM, Word2Vec). Applied to InsuranceQA dataset and customer …☆16Oct 17, 2017Updated 9 years ago