Wheelodex » Projects » jax-scalify » jax_scalify-0.1.1-py3-none-any.whl

jax-scalify

View on PyPI — Reverse Dependencies (0)

0.1.1 jax_scalify-0.1.1-py3-none-any.whl

Wheel Details

Project: jax-scalify
Version: 0.1.1
Filename: jax_scalify-0.1.1-py3-none-any.whl
Download: [link]
Size: 36009
MD5: 612eeec8e06641f7a1adcbda120173f0
SHA256: a93fd90cffd301294df650b1c5eabfd29551c86151674477bdb11d1c2b94f314
Uploaded: 2024-07-17 12:07:12 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: jax_scalify
Version: 0.1.1
Summary: JAX Scalify: end-to-end scaled arithmetic.
Author-Email: Paul Balanca <paulb[at]graphcore.ai>
Project-Url: Homepage, https://github.com/graphcore-research/jax-scalify/#readme
Project-Url: Bug Reports, https://github.com/graphcore-research/jax-scalify/issues
Project-Url: Source, https://github.com/graphcore-research/jax-scalify/
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: chex (>=0.1.6)
Requires-Dist: jax (>=0.3.16)
Requires-Dist: jaxlib (>=0.3.15)
Requires-Dist: ml-dtypes
Requires-Dist: numpy (>=1.22.4)
Requires-Dist: check-manifest; extra == "dev"
Requires-Dist: pytest; extra == "test"
Provides-Extra: dev
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4907 characters]

WHEEL

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

RECORD

Path Digest Size
jax_scalify/__init__.py sha256=3DnZZSHMKOw4MubIEKFy8LfvRa2h89tBna2bhQm1NdQ 301
jax_scalify/version.py sha256=PKIMyjdUACH4-ONvtunQCnYE2UhlMfp9su83e3HXl5E 411
jax_scalify/core/__init__.py sha256=Ww18_1LOqk3kaLF9i83cGiBxK70NWEj4ReMzPyDNcrE 815
jax_scalify/core/datatype.py sha256=Q5o8iv2AbUXkprktBINuUsaMzpwthBLICL1nvVbqe74 11571
jax_scalify/core/debug.py sha256=OxR4AAVrQqusPtVG4-8iJg22MuVwXBzowqkvL-lM5Do 2524
jax_scalify/core/interpreters.py sha256=t_-10TAue07LZS_mPbZ-4tjq1bkgbwJp4XYpoftzNNs 22683
jax_scalify/core/pow2.py sha256=ZKTydkXGjZeLmbU4h9YXFzh2AZciyzgbxREGWwNb6_A 5964
jax_scalify/core/typing.py sha256=m7JqRm25dfJiv7Fp7JPyQTBDDfMLr3qR2Nii_VW8RXM 1107
jax_scalify/core/utils.py sha256=wlqvdBm9zr0VgPVxBK3qPaTTIEG21HG01S_2Ahj7cZk 1563
jax_scalify/lax/__init__.py sha256=cCRubdObMvhgzxNhLPEI-J48vzRpFwlmcJGa766t3UA 345
jax_scalify/lax/base_scaling_primitives.py sha256=H2Nx2FS0nj-wRUjsU4S7NjQh4ytaB6nYQuev7Bn2NA4 8759
jax_scalify/lax/scaled_ops_common.py sha256=aZfl7NCmSGkRg-sTUS_v1kr35dHUH6Jyo7ZGAjJ4Aks 12210
jax_scalify/lax/scaled_ops_l2.py sha256=VLhI5ok3Et5oLKjeiiODIT0dbOIpTo2ZyHKDTAgA6_E 7620
jax_scalify/ops/__init__.py sha256=ld13BmBZQewVSndU7x887-elo0RwciloouhwcqneRf8 505
jax_scalify/ops/cast.py sha256=qrf2gZUT9wLHwMz2OlEBiHoXiW0DEX4z3zPvj_Wu7fU 1600
jax_scalify/ops/debug.py sha256=8ygCxKwGQ6-rcAHCTdXBhuV0rbQEMvWSxsTGm5irzfA 974
jax_scalify/ops/rescaling.py sha256=EvMD3Nr5i5n-0i5UVQWmeeXyoqzdojKWdf3sAp2kGjY 2984
jax_scalify/ops/utils.py sha256=PzbsSYt0n5bJRXHGezEpFPODlnWKOX4LMxNl2lQ1g_k 812
jax_scalify/tree/__init__.py sha256=rrp7PQLGUebyBGCLNuYHwj41NtTZjYt3SEejx2DEtjE 150
jax_scalify/tree/tree_util.py sha256=ARoc3fW91lR6va8ENnPj9MxK5AunIbuozGSfhlyst8Q 3911
jax_scalify-0.1.1.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax_scalify-0.1.1.dist-info/METADATA sha256=M8sibFxF-61hdXquhmeepOojXY5fh_siI7mPYLz1Yn4 6029
jax_scalify-0.1.1.dist-info/WHEEL sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk 91
jax_scalify-0.1.1.dist-info/top_level.txt sha256=Uas7QURuSWYVu7DPM99rt8mYE8Ul4P0ekrgSEScMFW8 12
jax_scalify-0.1.1.dist-info/RECORD — —

top_level.txt

jax_scalify