tf2jax

View on PyPIReverse Dependencies (1)

0.3.6 tf2jax-0.3.6-py3-none-any.whl

Wheel Details

Project: tf2jax
Version: 0.3.6
Filename: tf2jax-0.3.6-py3-none-any.whl
Download: [link]
Size: 97893
MD5: 79b6a4b29c6f16d3f84b23c02b9b408c
SHA256: 42c17bdefed405dfb3b87c3fe14107ee356a3a95e8637959f377036764488782
Uploaded: 2024-03-26 10:38:19 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: tf2jax
Version: 0.3.6
Summary: TF2JAX: Convert TensorFlow to JAX
Author: DeepMind
Author-Email: tf2jax-dev[at]google.com
Home-Page: https://github.com/deepmind/tf2jax
License: Apache 2.0
Keywords: jax tensorflow conversion translate
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Classifier: Topic :: Software Development :: Libraries
Requires-Python: >=3.7
Requires-Dist: absl-py
Requires-Dist: dm-tree (>=0.1.5)
Requires-Dist: jax (>=0.3.14)
Requires-Dist: jaxlib (>=0.1.37)
Requires-Dist: numpy (>=1.23.0)
Requires-Dist: tensorflow (>=2.8.0); sys_platform != "darwin" or platform_machine != "arm64"
Requires-Dist: tensorflow-macos (>=2.8.0); sys_platform == "darwin" and platform_machine == "arm64"
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 13871 characters]

WHEEL

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

RECORD

Path Digest Size
tf2jax/__init__.py sha256=1IMsb1WcgOFR3ofdFqP-sb-UZFmiEzFHv4696Np1rKw 1500
tf2jax/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tf2jax/tf2jax_test.py sha256=MT0W1AAAtsDQYCKX3f5C1a7h_x-4esyJjFEYrJi3US4 976
tf2jax/_src/__init__.py sha256=29SF9Wtyw1Y2FnYIjeBdNYg397L114-Wh5rYSegg6q0 696
tf2jax/_src/config.py sha256=DUI0yWoZJCOckIOYymfW8H5vCwbEMqT2fGSjPQRIgy8 2279
tf2jax/_src/linalg_ops_test.py sha256=XyuGAECxdTai-_-0V0Sn4TPLQ8RxkJvoMx7qqxUZlzc 10700
tf2jax/_src/numpy_compat.py sha256=hPIMv0odbE8Oz4_Sn60jDo67E5tNdXFZAqhmLulbIK0 8693
tf2jax/_src/numpy_compat_test.py sha256=Bdq20nz-t-w2NDxV65Twdar7U5wC_6Obx6YftOlxo68 2146
tf2jax/_src/ops.py sha256=p99T0kFBOKKHN9ZxU_MoVaa70fRlsoOLDedQUrKTOOM 90031
tf2jax/_src/ops_test.py sha256=6lDKAt5f5J-xS6FqG_hfVGv4um0cSYRordPSewGCzcU 78028
tf2jax/_src/roundtrip_test.py sha256=ejXXTmaRuopai5f4-59XV95soTqlkuT8-v0l_v_QTuk 42690
tf2jax/_src/sharding_test.py sha256=V1F5lWmq735biH1GouU-PXUHrSVJTvQpufWZRV9ptFo 5156
tf2jax/_src/test_util.py sha256=a5-3z7qX8X71Iz92qTVUkCL4iQ8vUiABgK4ZPAIcJFI 1617
tf2jax/_src/tf2jax.py sha256=awrvgrorploemKWYfb6HpBdBtRlVvLPCenyKxOr3k2E 52891
tf2jax/_src/tf2jax_test.py sha256=4iRS6FL6A5o5B5n9ROpkTrMx4ucrTcoPgPx7yBApxU4 20233
tf2jax/_src/utils.py sha256=hJGIKdf5BROsu_lJpiZJDsUNfeWLRIamgO58Ze81Eiw 1974
tf2jax/_src/xla_utils.py sha256=DWEKPhivqfFg1xTIjKCMIBpW5bZsIYR_nYLqlborCSk 4291
tf2jax/experimental/__init__.py sha256=29SF9Wtyw1Y2FnYIjeBdNYg397L114-Wh5rYSegg6q0 696
tf2jax/experimental/mhlo.py sha256=rvtIN0gtSJKAmZ6jD3wXt_nra2N6-TJ-nq9ZYFv3mnk 8357
tf2jax/experimental/mhlo_test.py sha256=JXtJHzkDV4P9lRrWsMBUPMp-dhFteSGtkoiQVKh3I_I 5597
tf2jax/experimental/ops.py sha256=BFxhq9SR13CGUgw3Ju1qhsUlQOAZtdpdngwBzz6FQKQ 7839
tf2jax/test_data/custom_gradient_cubed/fingerprint.pb sha256=vYoyo-4CMx1Gl77hfdhefiV7jbV6u0uChCIecbnlqa0 57
tf2jax/test_data/custom_gradient_cubed/saved_model.pb sha256=Lghk-vX2iiZy8P9YjatWQRM9VwK2iCMidzo3H9Oo9qo 9007
tf2jax/test_data/custom_gradient_cubed/variables/variables.data-00000-of-00001 sha256=JxKNj_AtLXFQLtcFKS59WxYqipCuPDeaHBoCcQjfz8U 57
tf2jax/test_data/custom_gradient_cubed/variables/variables.index sha256=KKlq2EeSZfR4QRElXjCOx-yvR3yfBe4fLasWl6A9q54 144
tf2jax-0.3.6.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
tf2jax-0.3.6.dist-info/METADATA sha256=KCO_5Mjn6vhjwO0A5Ve59Yh9FwHcCwsap3WwjUTJeHk 15198
tf2jax-0.3.6.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
tf2jax-0.3.6.dist-info/top_level.txt sha256=xL44xhHTeVVZcq8GP7hjQziWTT2DqUSKsMTQADgocq8 7
tf2jax-0.3.6.dist-info/RECORD

top_level.txt

tf2jax