nikopj / FlashAttention.jlView external linksLinks
Julia implementation of flash-attention operation for neural networks.
☆11May 31, 2023Updated 2 years ago
Alternatives and similar repositories for FlashAttention.jl
Users that are interested in FlashAttention.jl are comparing it to the libraries listed below
Sorting:
- ☆15Mar 21, 2025Updated 10 months ago
- Integrating Neural Ordinary Differential Equations, the Method of Lines, and Graph Neural Networks☆18Oct 29, 2023Updated 2 years ago
- Experimental features for Flux.jl☆24Jan 5, 2025Updated last year
- Julia interactive interface for TinyMPC☆11Aug 28, 2025Updated 5 months ago
- Modular PyTorch (Lightning) implementation of Diffusion Probabilistic Models☆22Mar 26, 2023Updated 2 years ago
- Immutables as mutables, mutables as immutables.☆23Jan 25, 2026Updated 3 weeks ago
- A lightweight package for the transformer deep learning architecture in Julia☆43Mar 8, 2025Updated 11 months ago
- ☆11Aug 27, 2021Updated 4 years ago
- Machine learning from scratch in Julia☆33Feb 26, 2025Updated 11 months ago
- GPU integrations for Dagger.jl☆54Jul 1, 2025Updated 7 months ago
- Reusable functionality for defining custom attention/transformer layers.☆55Nov 24, 2025Updated 2 months ago
- CUDA integration for the NNlib API☆14Feb 5, 2023Updated 3 years ago
- Convenience wrapper for PyAMG library☆10Oct 27, 2025Updated 3 months ago
- A fast compressor/decompressor☆16Nov 18, 2024Updated last year
- Interface designs for enforcing static computations in array functions with Julia☆15Updated this week
- ☆15Jan 18, 2026Updated 3 weeks ago
- A new lightweight package for Deep Learning with Julia☆14Jan 29, 2024Updated 2 years ago
- Optimisers.jl defines many standard optimisers and utilities for learning loops.☆93Dec 13, 2025Updated 2 months ago
- Reactive, "soft real-time" programming tools for Julia.☆18Nov 30, 2023Updated 2 years ago
- Stanford PhD Thesis☆18Jan 11, 2023Updated 3 years ago
- Julia Interface to https://github.com/compiler-research/CppInterOp☆19May 10, 2025Updated 9 months ago
- Memory efficient one-hot array encodings☆21May 29, 2025Updated 8 months ago
- A comprehensive collection of 35+ recurrent neural network layers for Flux.jl☆29Oct 30, 2025Updated 3 months ago
- Implementation of the TintiNet architecture for Julia☆17Mar 1, 2024Updated last year
- Common hyperparameter scheduling for ML☆38Dec 3, 2024Updated last year
- Data structures for graph neural network☆18May 25, 2024Updated last year
- No need to train, he's a smooth operator☆45Dec 15, 2025Updated 2 months ago
- A flexible neural net training library inspired by fast.ai☆122Dec 25, 2024Updated last year
- Unofficial Julia bindings for logging experiments to wandb.ai☆89Sep 27, 2024Updated last year
- Parameterise all the things☆123Dec 15, 2025Updated 2 months ago
- Simple chains☆242Feb 1, 2026Updated 2 weeks ago
- This Julia package provides the testitem macro☆24Nov 6, 2025Updated 3 months ago
- A Julia wrapper for the NVIDIA Collective Communications Library.☆29Dec 1, 2025Updated 2 months ago
- Nonlinear Distributionally-Robust MPC via Risk-Sensitive Optimal Control in Julia☆21Aug 9, 2021Updated 4 years ago
- ☆21Jul 29, 2023Updated 2 years ago
- Line search methods for optimization and root-finding☆144Jan 8, 2026Updated last month
- Limited-Memory Factorization of Symmetric Matrices☆24Jun 8, 2025Updated 8 months ago
- Computing mixed-strategy Nash Equilibria for games involving multiple players☆25Jan 16, 2025Updated last year
- Value iteration solver for MDPs☆21Nov 16, 2025Updated 2 months ago