Wheelodex » Projects » haliax

haliax

View on PyPI — Reverse Dependencies (1)

1.3 haliax-1.3-py3-none-any.whl

Wheel Details

Project: haliax
Version: 1.3
Filename: haliax-1.3-py3-none-any.whl
Download: [link]
Size: 96876
MD5: e6446255277e2092b6b9a86388ee3256
SHA256: 9cc4d2258d0596de4c56f015502ed491101854e54d72fb4d08d678d8476e8b97
Uploaded: 2024-04-08 05:27:26 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: haliax
Version: 1.3
Summary: Named Tensors for Legible Deep Learning in JAX
Author-Email: David Hall <dlwh[at]cs.stanford.edu>
Project-Url: Homepage, https://github.com/stanford-crfm/haliax
Project-Url: Bug Tracker, https://github.com/stanford-crfm/haliax/issues/
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: MacOS :: MacOS X
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.10
Requires-Dist: equinox (>=0.10.6)
Requires-Dist: jaxtyping (>=0.2.20)
Requires-Dist: jmp (>=0.0.4)
Requires-Dist: mkdocs-include-markdown-plugin; extra == "dev"
Requires-Dist: mkdocs-literate-nav (>=0.6.0); extra == "dev"
Requires-Dist: mkdocs-macros-plugin (>=0.7.0); extra == "dev"
Requires-Dist: mkdocs-material (>=7.3.3); extra == "dev"
Requires-Dist: mkdocs (>=1.4.3); extra == "dev"
Requires-Dist: mkdocstrings-python (>=1.1.2); extra == "dev"
Requires-Dist: mkdocstrings (>=0.22.0); extra == "dev"
Requires-Dist: mypy (>=0.910); extra == "dev"
Requires-Dist: pygments; extra == "dev"
Requires-Dist: pymdown-extensions; extra == "dev"
Requires-Dist: pytest (>=7.4.0); extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6093 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.22.5
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
haliax/__about__.py sha256=X5x45yf0U-A9-P3NtDk6aMAXwiInEYM9rXxWXG1rU10 27
haliax/__init__.py sha256=u8MqucR8tzN4dP2P24ELkO3j9zRB-N76fDXGJSCrZEE 28790
haliax/axis.py sha256=J7K3_2WZfJAIwi1AgL8Mo7twIEUTPkwA3Hb_KkgRf7w 17892
haliax/core.py sha256=w0ZgXGMHRKPSYawxSEfiq2jlWTk2loUZA4ni9bexeYo 55808
haliax/debug.py sha256=0qEGgIsEw3Jkp40oxgBAjD0Ps-cxi3QNP42uMRGlM84 3612
haliax/hof.py sha256=jDcii2IwAfhoNYTCYsdeS8vpSKmtzioPKZ6LBAVuWn8 18205
haliax/jax_utils.py sha256=-qiNgxSvN2de3ZK1luPqfj_PO_mk6mKGPMM0f89Ky9I 6298
haliax/ops.py sha256=Rrsz5GiTL8SZc8g8i7en0OkfH_HTEMMhVnWVh696YSU 5138
haliax/partitioning.py sha256=jMzDVswA1dhyMx-nT6YruUbLMzavyB-gbmB7RVwoi3Y 22548
haliax/quantization.py sha256=0Un7Pr34BWbnY0oMGkjxUuYRrhIBi6_DMc0DlBR56_o 9442
haliax/random.py sha256=Qasyt7CSqbcUC9-Q715dGsctpZS0qzHKJ-9smUzoP78 13077
haliax/specialized_fns.py sha256=reaG54m6BFTDEgInpylEaZ1YRnKlwuwLikEw7Jv3vDQ 1470
haliax/tree_util.py sha256=315b0RaNrcCcQ7B0eid09mLwNFmeisnfo3zpqJdKNow 4381
haliax/types.py sha256=KShYKvXtaCA63OdDlolFhcCQfZD0owZugg4mX3K5RDQ 834
haliax/util.py sha256=9md_yCTVH97fPcl4Cin7aAaKT4HaI1VnF0Zn7DEJT8A 2574
haliax/wrap.py sha256=GRdo-0QuiW-hNPS85_8MUIFQ3AupYmG65JumWbZHfMo 4564
haliax/_src/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
haliax/_src/compile_utils.py sha256=TJ96sX8gtx_7hUaCuYKrWojTTdXUIjtDMogr4FmKkPE 2696
haliax/_src/dot.py sha256=TVEgwPstXSFv1-TvORUSA_fM4lz1YrAAy7fJzH26KLc 6265
haliax/_src/einsum.py sha256=unGwChKsWKBd8vcAzRHFRs4_KKoBdswZ-BDZSnsRnSM 11696
haliax/_src/fp8.py sha256=wtyOK7TrXgg4UlwydDklCMXvkbGMoixB3mKOk3IIJvM 4951
haliax/_src/parsing.py sha256=YLGOl1_BOPcrwC_oo-STK4mD6xy4Kd0S5UzUGUuZ1Ps 10691
haliax/_src/rearrange.py sha256=SWZAIMSrqQzqwZDxrMk25q6IOaK4PyDUEm-VOSe0N18 19498
haliax/_src/util.py sha256=pXizGeJfRcx7QA1YVAxvru2EnMGIXAhZdcPGkwatP_M 1566
haliax/nn/__init__.py sha256=yZy-fIfVhWhbbUpvYBbtgYnM9a8fGFcXBvbSio7pY4E 2858
haliax/nn/activations.py sha256=9h1uJ2pUWmb_OsRhqUGUJ6s95WZgxABDx7YWFD2tn5A 1723
haliax/nn/attention.py sha256=QInFORjeMyO9sxkr9XoU8ayy08QD2Qu6jXc2gW-GzOE 11718
haliax/nn/conv.py sha256=ozdvYBnb20CDz_xMKiXdpOW0ZMBZKlwadx21XafFw3E 15131
haliax/nn/dropout.py sha256=2BETs0MeHMdzUuGof3vdVRZSwC7vQwKP9BT-Ix0lDRY 3682
haliax/nn/embedding.py sha256=a66bQNoPts6SPRxzBd_dTtVEGUljO-aJxfISA5VA1A8 1356
haliax/nn/linear.py sha256=VB7CdpYY7ZcsWiA0zVGJttBQhT9lqp1h_0N9W1VEnXM 2418
haliax/nn/loss.py sha256=qdrtaXKpV8Q6YOQyNz19Al4jyZLfmL5QhzNyA9z8mdA 4011
haliax/nn/mlp.py sha256=g5YqvwxhaqxaZiFFymWJiKsUasyeQ2_qCOxCvU1po4I 3396
haliax/nn/normalization.py sha256=4e1rpVvZKW9w5eUsn8j0tBF6lAIs5YljsvOE_HaoOD8 2652
haliax/nn/pool.py sha256=Wv1bygx_aMFp8OWi7q3E4X4XgNB-UtIpVah19foE2mE 8513
haliax/nn/scan.py sha256=gE10cBL_0poKUOw3ryj_i6sqvFmmN4Ar7P3xuM7l-u8 10226
haliax-1.3.dist-info/METADATA sha256=6DkJIlIoPd3EEMZUNGf8HqnmTPohCxyIzgFP9_OW2XQ 7453
haliax-1.3.dist-info/WHEEL sha256=as-1oFTWSeWBgyzh0O_qF439xqBe6AbBgt4MfYe5zwY 87
haliax-1.3.dist-info/licenses/LICENSE sha256=bJiay7Nn5SHQ2n_4ZIT3AE0W1RGq4O7pxOApgBsaT64 11349
haliax-1.3.dist-info/RECORD — —