YichengDWu / FlashAttention.jl
Julia implementation of the Flash Attention algorithm
☆18Updated last year
Related projects: ⓘ
- Simple, blazing fast, transformer components.☆18Updated last year
- Tools for visualizing Julia IR☆44Updated 3 months ago
- Pruning framework and methods for Flux☆27Updated last year
- Curated list of high-quality operators for deep learning in Julia☆39Updated 2 years ago
- Julia interface for dlpack☆44Updated 3 months ago
- An experimental simple method overlay mechanism for Julia☆28Updated 2 months ago
- ☆24Updated 4 years ago
- Using runtime-free macro packages as dev-only dependencies.☆22Updated last year
- AD backends for LogDensityProblems.jl.☆12Updated this week
- Julia preferences for humans☆32Updated last year
- Using compiler magics to apply fastmath flags in a scoped fashion☆9Updated last year
- A Julia wrapper for the NVIDIA Collective Communications Library.☆24Updated last month
- Pass loop info to LLVM☆21Updated last year
- ☆35Updated 2 years ago
- Generate precompile directives by reading method signatures☆39Updated 2 years ago
- The Code Tracer☆32Updated 6 months ago
- Composable nonblocking and synchronization programming framework☆23Updated last month
- Methodwise Memoization for Julia☆22Updated 2 years ago
- Diffusion models for image generation in Julia☆9Updated 7 months ago
- ☆20Updated 2 years ago
- Memory efficient one-hot array encodings☆18Updated 7 months ago
- Accelerate your ML research using pre-built Deep Learning Models with Lux☆26Updated last week
- Quaternions for Julia☆19Updated last week
- Julia package for creating composable invariants with helpful error messages☆25Updated last year
- 🥞☆28Updated 11 months ago
- Proof of concept for type system with unions, intersections and complements.☆13Updated last year
- EfficientNet implementation in Julia☆20Updated 2 years ago
- Autosuggestions for function keywords☆19Updated 3 months ago
- More insights about your packages dependencies☆23Updated last year
- Multiple Hilbert curve algorithms in two dimensions, N-dimensions, and with unequal side lengths, in native Julia.☆21Updated 8 months ago