PyPI page
Home page
Author:
None
License:
MIT License
Copyright (c) 2025 Phil Wang
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated document...
Summary:
Native Sparse Attention
Latest version:
0.2.3
Required dependencies:
einops
|
einx
|
jaxtyping
|
local-attention
|
rotary-embedding-torch
|
torch
Optional dependencies:
pytest
|
tqdm
|
wandb
Downloads last day:
692
Downloads last week:
855
Downloads last month:
944