jax-triton

View on PyPIReverse Dependencies (0)

0.2.0 jax_triton-0.2.0-py3-none-any.whl

Wheel Details

Project: jax-triton
Version: 0.2.0
Filename: jax_triton-0.2.0-py3-none-any.whl
Download: [link]
Size: 27053
MD5: acb871f67693a6020f1e427cf879ec00
SHA256: 6db5f6147327f2462c35becab612d415cf9ee70c96e43020c3c7ea873eb8576e
Uploaded: 2024-10-23 11:21:07 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax-triton
Version: 0.2.0
Summary: JAX + OpenAI Triton integration
Requires-Python: >=3.10
Requires-Dist: absl-py (>=1.4.0)
Requires-Dist: jax (>=0.4.34)
Requires-Dist: triton (>=3.1)
Requires-Dist: pytest; extra == "tests"
Provides-Extra: tests
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 2643 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.2.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jax_triton/__init__.py sha256=uJiRp76Pz7f3Hz41vCk5rgu5keS2pNUX2tCX8J3KcOA 1631
jax_triton/triton_lib.py sha256=Eldl441kERFeaK435EuLl8vMiRgXTvQ0bOGlFWfQSG4 26557
jax_triton/utils.py sha256=gLH2XcRClKUJQH13i3FSjOyx8FChl1MW4Vylh_2-CbM 867
jax_triton/version.py sha256=7_RfcgEyv_0IYjhy-mq7EMP7l81obgj_O2V-AEnR0HY 674
jax_triton/experimental/fusion/__init__.py sha256=KP18QHnGBOP17TZSQ5Sc2Vba_7ZrirJCVy5-r7lGiYw 797
jax_triton/experimental/fusion/fusion.py sha256=wzA-NnuSf3hqe3BYGQ9snwKpPWe_2F3LRc9N9bGMypg 12092
jax_triton/experimental/fusion/jaxpr_rewriter.py sha256=hPqwpHf2F25Z4iQxJa8mr-_iuHcl_Abi0JrJYS2FAyk 9136
jax_triton/experimental/fusion/lowering.py sha256=K2JNWLIjegmoVMeuTwL4rfnk5vLnC1VBOs28xhazM2s 14588
jax_triton-0.2.0.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax_triton-0.2.0.dist-info/METADATA sha256=kno_ZloNgJqTCoiPFJWBvQrWXFhbBMAKWWmuohFGJ7Y 2975
jax_triton-0.2.0.dist-info/WHEEL sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA 91
jax_triton-0.2.0.dist-info/top_level.txt sha256=L1fwlmYQtpXGcX5MjDoRqbShO8E4Fow-jxVSGZPyge8 11
jax_triton-0.2.0.dist-info/RECORD

top_level.txt

jax_triton