cubek-attention-0.1.1
- cubek-attention 0.1.1
- Permalink
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Repository
- crates.io
- Source
- Owners
- nathanielsimard
- Dependencies
- bytemuck ^1.16.1 normal
  cubecl ^0.9.0 normal
  cubecl-common ^0.9.0 normal
  cubek-matmul ^0.1.1 normal
  cubek-random ^0.1.1 normal
  half ^2.5 normal
  serde ^1.0.204 normal
  cubecl ^0.9.0 dev
  cubek-test-utils ^0.1.1 dev
  pretty_assertions ^1.4 dev
  trybuild ^1 dev
- Versions
- 24.61% of the crate is documented
Platform
Feature flags

docs.rs

Rust

Crate cubek_attention

cubek_attention0.1.1

All Items

Crate Items

Modules

Crate cubek_attention

Modules§

components: Components for matrix multiplication
definition
launch
routines: Contains attention kernels