☆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:
- 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
- ☆13Apr 16, 2025Updated last year
- 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.
- ☆35Sep 6, 2025Updated 11 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆174Nov 11, 2025Updated 9 months ago
- Smart reproducible analytical pipeline inspection☆21Feb 13, 2026Updated 6 months ago
- JAX Implementations of Descript Audio Codec and EnCodec☆38Mar 30, 2025Updated last year
- Particle filtering in JAX☆17Jun 12, 2026Updated 2 months ago
- List of new Project Fundraising Opportunities for NumFOCUS Sponsored Projects☆14May 14, 2026Updated 3 months ago
- GGUF parser in Python☆28May 1, 2026Updated 3 months ago
- Custom kernels in Triton language for accelerating LLMs☆27Apr 5, 2024Updated 2 years ago
- 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
- 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.
- 3D Gaussian Splatting in JAX☆76May 30, 2024Updated 2 years ago
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Jul 7, 2026Updated last month
- cuda implementation of depthwise conv3d☆23Jul 14, 2021Updated 5 years ago
- Pytorch-like dataloaders for JAX.☆105Dec 16, 2025Updated 8 months ago
- 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 2 years ago
- Sparse Autoencoder Training Library☆58May 1, 2025Updated last year
- Next Generation Experimental Tracking for Machine Learning Operations☆17Updated this week
- 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.
- Tokamax: A GPU and TPU kernel library.☆265Updated this week
- a Jax quantization library☆130Updated this week
- ☆19Dec 4, 2025Updated 8 months ago
- 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☆276Jul 4, 2026Updated last month
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆372Aug 16, 2026Updated last week
- VTSCat is the catalogue of all published observations of the VERITAS collaboration.☆12Jun 16, 2026Updated 2 months ago
- An unnecessarily tiny and minimal implementation of GPT-2 in NumPy.☆11Feb 12, 2023Updated 3 years ago
- 北京大学跑步软件PKURUNNER破解☆21May 13, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Convert StableHLO models into Apple Core ML format☆22Updated this week
- 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"☆85Dec 18, 2025Updated 8 months ago
- Mxnet implementation of an ICLR 2018 paper: A new method of region embedding for text classification.☆10Oct 14, 2018Updated 7 years ago
- Reversal Curse Experiment☆15Sep 24, 2023Updated 2 years ago
- Personalized Fashion Compatibility Modeling via Metapath-guided Heterogeneous Graph Learning.☆16Nov 7, 2022Updated 3 years ago