Wheelodex » Projects » wrap-torch2jax

wrap-torch2jax

View on PyPI — Reverse Dependencies (0)

0.5.0

wrap_torch2jax-0.5.0-py3-none-any.whl

Wheel Details

Project:	wrap-torch2jax
Version:	0.5.0
Filename:	wrap_torch2jax-0.5.0-py3-none-any.whl
Download:	[link]
Size:	20334
MD5:	`fe6b4a5796e9883f867ac25a098c00fa`
SHA256:	`602655f1a3e05ea4ea04a08e6e616ebd3184a13f3b4798816ccbcdefe6035822`
Uploaded:	2024-11-29 18:45:58 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version:	2.1
Name:	wrap_torch2jax
Version:	0.5.0
Summary:	Wrap your PyTorch for JAX! This package allows no-copy PyTorch calling from JAX under both eager execution and JIT.
Author-Email:	Robert Dyro <robert.dyro[at]gmail.com>
Project-Url:	Homepage, https://github.com/rdyro/torch2jax
Project-Url:	Bug Tracker, https://github.com/rdyro/torch2jax
Classifier:	Programming Language :: Python :: 3
Classifier:	License :: OSI Approved :: MIT License
Classifier:	Operating System :: OS Independent
Requires-Python:	>=3.7
Requires-Dist:	numpy
Requires-Dist:	jax
Requires-Dist:	torch
Description-Content-Type:	text/markdown

[Description omitted; length: 11728 characters]

WHEEL

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

RECORD

Path	Digest	Size
wrap_torch2jax/__init__.py	sha256=G6ZDgdYoW9bx4eGv-nmzReQysfpy18Ft-NTx9_qP-Cg	457
wrap_torch2jax/api.py	sha256=4MG1EgrJHJG7RSFzNPFzyAOJbDDBNqz7IxGUpy6hpeg	6616
wrap_torch2jax/compat.py	sha256=IAzmeaNDJnt-INwULQkexAwtiB4RmXKVjCyKUvyiKyg	59
wrap_torch2jax/compile.py	sha256=nbrGB-4t-NGgfS5gr48EPQGbjw2xWTQZO2_hnIk3CHQ	3468
wrap_torch2jax/dlpack_passing.py	sha256=Tr741DLe9b9GoN7-NqeA7k_4qAvM-W0_IE6F0y2eqxQ	2390
wrap_torch2jax/gradients.py	sha256=0c_khPygdDjhv0Do1XfLdWfBFg2aAOytbdsLCf5N6Es	9513
wrap_torch2jax/utils.py	sha256=12JCxQb6sAjVtfoiAdiRt79nF1HkpkHPcRDdTd3PRRU	4583
wrap_torch2jax/cpp/common.cpp	sha256=goQX3-zL5ZLw0fXNd51atzalpGZ8T1u48DKrS_FXgmI	4853
wrap_torch2jax/cpp/cpu_impl.cpp	sha256=l8apwxpcyWOurDURR5YQvyPmjChs8YyzbJ7Gs4TqqQo	1052
wrap_torch2jax/cpp/cpu_impl.h	sha256=6P3Ph_tz47O8B69MIkBH2kNiKzx1P4-W9YDVk67r7U8	261
wrap_torch2jax/cpp/gpu_impl.cu	sha256=XLrZ3Mq9sLlZYXPgWAU6oNmwadDMgTCCAI8e57UFBsU	1363
wrap_torch2jax/cpp/gpu_impl.h	sha256=ou24WXvnc4SS9hzC4fRJwlwba2MuHRZIYSLHm46VYC8	326
wrap_torch2jax/cpp/main.cpp	sha256=1Koj-wJS_YA6wds6QcC50IqkSHkeqiD_fDhKqMcAQ4M	171
wrap_torch2jax/cpp/main.cu	sha256=c6bzooTZwUZa4L2USvrgA2N-T9W_wj_o31C9vIcJK4E	241
wrap_torch2jax/cpp/main.h	sha256=VmnP-HvAXt0RFGWA_9IPfTUYUMj_w4QavsTmVYIU86s	2611
wrap_torch2jax-0.5.0.dist-info/METADATA	sha256=U8szJgCzer2q1lyRLqUARJ40WE1YNKCL0b-oTPGnoc0	12351
wrap_torch2jax-0.5.0.dist-info/WHEEL	sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U	91
wrap_torch2jax-0.5.0.dist-info/top_level.txt	sha256=4Le98i_tR89N39qK8p4W0Vud-ZZZetVsC6oubw8AwLQ	15
wrap_torch2jax-0.5.0.dist-info/RECORD	—	—

top_level.txt

wrap_torch2jax