Change the repository type filter
All
Repositories list
11 repositories
flash-attention-jax
Publicnanobind
Publichf-transformers
PublicTransformerEngine
Publictransformers
Publicequinox
Public archiveElegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/einops
Publicflash-attention
Publicflax
Public