wrap-torch2jax
Wrap your PyTorch for JAX! This package allows no-copy PyTorch calling from JAX under both eager execution and JIT.
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install wrap-torch2jax
Dependencies
- None
Releases
| Version | Released | Bookworm Python 3.11 |
Trixie Python 3.13 |
Files |
|---|---|---|---|---|
| 0.9.0 | 2026-09-26 | |||
|
|
||||
| 0.8.3 | 2026-07-12 | |||
|
|
||||
| 0.8.1 | 2026-04-12 | |||
|
|
||||
| 0.8.0 | 2026-04-12 | |||
|
|
||||
| 0.7.2 | 2026-04-11 | |||
|
|
||||
| 0.7.1 | 2026-02-15 | |||
|
|
||||
| 0.7.0 | 2025-07-23 | |||
|
|
||||
| 0.5.0 | 2024-11-29 | |||
|
|
||||
| 0.4.11 | 2024-08-12 | |||
|
|
||||
| 0.4.10 | 2024-04-29 | |||
|
|
||||
| 0.4.7 | 2023-11-13 | |||
|
|
||||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue