stackkernels 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (36) hide show
  1. stackkernels/__init__.py +0 -0
  2. stackkernels/hub/__init__.py +0 -0
  3. stackkernels/hub/registry.py +0 -0
  4. stackkernels/jax/__init__.py +0 -0
  5. stackkernels/jax/_utils.py +0 -0
  6. stackkernels/jax/activation.py +0 -0
  7. stackkernels/jax/attention.py +0 -0
  8. stackkernels/jax/loss.py +0 -0
  9. stackkernels/jax/norm.py +0 -0
  10. stackkernels/torch/__init__.py +0 -0
  11. stackkernels/torch/_utils.py +0 -0
  12. stackkernels/torch/activation.py +0 -0
  13. stackkernels/torch/attention.py +0 -0
  14. stackkernels/torch/dropout.py +0 -0
  15. stackkernels/torch/embedding.py +0 -0
  16. stackkernels/torch/loss.py +0 -0
  17. stackkernels/torch/norm.py +0 -0
  18. stackkernels/triton/__init__.py +0 -0
  19. stackkernels/triton/_heuristics.py +0 -0
  20. stackkernels/triton/activation/gelu.py +0 -0
  21. stackkernels/triton/activation/silu.py +0 -0
  22. stackkernels/triton/activation/swiglu.py +0 -0
  23. stackkernels/triton/attention/causal_mask.py +0 -0
  24. stackkernels/triton/attention/flash_attn.py +0 -0
  25. stackkernels/triton/attention/paged_attn.py +0 -0
  26. stackkernels/triton/attention/varlen_attn.py +0 -0
  27. stackkernels/triton/dropout/dropout.py +0 -0
  28. stackkernels/triton/loss/cross_entropy.py +0 -0
  29. stackkernels/triton/norm/fused_norm.py +0 -0
  30. stackkernels/triton/norm/layer_norm.py +0 -0
  31. stackkernels/triton/norm/rms_norm.py +0 -0
  32. stackkernels/triton/positional/rope.py +0 -0
  33. stackkernels-0.1.0.dist-info/METADATA +39 -0
  34. stackkernels-0.1.0.dist-info/RECORD +36 -0
  35. stackkernels-0.1.0.dist-info/WHEEL +4 -0
  36. stackkernels-0.1.0.dist-info/licenses/LICENSE +0 -0
stackkernels/__init__.py ADDED
File without changes
stackkernels/hub/__init__.py ADDED
File without changes
stackkernels/hub/registry.py ADDED
File without changes
stackkernels/jax/__init__.py ADDED
File without changes
stackkernels/jax/_utils.py ADDED
File without changes
stackkernels/jax/activation.py ADDED
File without changes
stackkernels/jax/attention.py ADDED
File without changes
stackkernels/jax/loss.py ADDED
File without changes
stackkernels/jax/norm.py ADDED
File without changes
stackkernels/torch/__init__.py ADDED
File without changes
stackkernels/torch/_utils.py ADDED
File without changes
stackkernels/torch/activation.py ADDED
File without changes
stackkernels/torch/attention.py ADDED
File without changes
stackkernels/torch/dropout.py ADDED
File without changes
stackkernels/torch/embedding.py ADDED
File without changes
stackkernels/torch/loss.py ADDED
File without changes
stackkernels/torch/norm.py ADDED
File without changes
stackkernels/triton/__init__.py ADDED
File without changes
stackkernels/triton/_heuristics.py ADDED
File without changes
stackkernels/triton/activation/gelu.py ADDED
File without changes
stackkernels/triton/activation/silu.py ADDED
File without changes
stackkernels/triton/activation/swiglu.py ADDED
File without changes
stackkernels/triton/attention/causal_mask.py ADDED
File without changes
stackkernels/triton/attention/flash_attn.py ADDED
File without changes
stackkernels/triton/attention/paged_attn.py ADDED
File without changes
stackkernels/triton/attention/varlen_attn.py ADDED
File without changes
stackkernels/triton/dropout/dropout.py ADDED
File without changes
stackkernels/triton/loss/cross_entropy.py ADDED
File without changes
stackkernels/triton/norm/fused_norm.py ADDED
File without changes
stackkernels/triton/norm/layer_norm.py ADDED
File without changes
stackkernels/triton/norm/rms_norm.py ADDED
File without changes
stackkernels/triton/positional/rope.py ADDED
File without changes
stackkernels-0.1.0.dist-info/METADATA ADDED
@@ -0,0 +1,39 @@
1
+ Metadata-Version: 2.5
2
+ Name: stackkernels
3
+ Version: 0.1.0
4
+ Summary: Fast, fused Triton kernels for transformer models — PyTorch and JAX backends.
5
+ Project-URL: Homepage, https://github.com/stackformer-labs/stackkernels
6
+ License: MIT
7
+ License-File: LICENSE
8
+ Requires-Python: >=3.9
9
+ Requires-Dist: torch<3.0,>=2.3
10
+ Requires-Dist: triton>=2.3
11
+ Provides-Extra: dev
12
+ Requires-Dist: pytest; extra == 'dev'
13
+ Requires-Dist: pytest-benchmark; extra == 'dev'
14
+ Provides-Extra: jax
15
+ Requires-Dist: jax-triton; extra == 'jax'
16
+ Requires-Dist: jax>=0.4; extra == 'jax'
17
+ Description-Content-Type: text/markdown
18
+
19
+ # stackkernels
20
+
21
+ Fast, fused Triton kernels for transformer models — built as the kernel backbone for [Stackformer](#).
22
+
23
+ Supports **PyTorch** and **JAX**, with a shared, framework-agnostic Triton core.
24
+
25
+ ## Status
26
+ 🚧 Early development. Build order: RMSNorm → LayerNorm → Flash Attention → Dropout → Activations (SiLU/GELU/SwiGLU) → Cross-Entropy → RoPE.
27
+
28
+ ## Install
29
+ ```bash
30
+ pip install stackkernels
31
+ ```
32
+
33
+ ## Quick start
34
+ ```python
35
+ from stackkernels.torch import RMSNorm
36
+ ```
37
+
38
+ ## License
39
+ MIT
stackkernels-0.1.0.dist-info/RECORD ADDED
@@ -0,0 +1,36 @@
1
+ stackkernels/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
+ stackkernels/hub/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ stackkernels/hub/registry.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
+ stackkernels/jax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
+ stackkernels/jax/_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
+ stackkernels/jax/activation.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
+ stackkernels/jax/attention.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
+ stackkernels/jax/loss.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
+ stackkernels/jax/norm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
+ stackkernels/torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
+ stackkernels/torch/_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
12
+ stackkernels/torch/activation.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
+ stackkernels/torch/attention.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
14
+ stackkernels/torch/dropout.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
+ stackkernels/torch/embedding.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
+ stackkernels/torch/loss.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
17
+ stackkernels/torch/norm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
18
+ stackkernels/triton/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
+ stackkernels/triton/_heuristics.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
20
+ stackkernels/triton/activation/gelu.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
21
+ stackkernels/triton/activation/silu.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
+ stackkernels/triton/activation/swiglu.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
23
+ stackkernels/triton/attention/causal_mask.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
+ stackkernels/triton/attention/flash_attn.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
25
+ stackkernels/triton/attention/paged_attn.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
26
+ stackkernels/triton/attention/varlen_attn.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
27
+ stackkernels/triton/dropout/dropout.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
+ stackkernels/triton/loss/cross_entropy.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
+ stackkernels/triton/norm/fused_norm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
+ stackkernels/triton/norm/layer_norm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
31
+ stackkernels/triton/norm/rms_norm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
32
+ stackkernels/triton/positional/rope.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
33
+ stackkernels-0.1.0.dist-info/METADATA,sha256=UW_kBsu_j1XuWJDnYd8VchG315rlTtQV3b-pjGa7Nig,1083
34
+ stackkernels-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
35
+ stackkernels-0.1.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
36
+ stackkernels-0.1.0.dist-info/RECORD,,
stackkernels-0.1.0.dist-info/WHEEL ADDED
@@ -0,0 +1,4 @@
1
+ Wheel-Version: 1.0
2
+ Generator: hatchling 1.32.4
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
stackkernels-0.1.0.dist-info/licenses/LICENSE ADDED
File without changes