├── .copier-answers.yml ├── .github └── workflows │ ├── build.yaml │ ├── docs.yaml │ └── publish.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── .python-version ├── LICENSE ├── README.md ├── docs ├── index.md └── reference │ └── torch_jax_interop.md ├── mkdocs.yaml ├── pyproject.toml ├── torch_jax_interop ├── __init__.py ├── conftest.py ├── docs_test.py ├── py.typed ├── to_jax.py ├── to_jax_module.py ├── to_jax_module_test.py ├── to_jax_test.py ├── to_torch.py ├── to_torch_module.py ├── to_torch_module_test.py ├── to_torch_test.py ├── types.py └── utils.py └── uv.lock /.copier-answers.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.copier-answers.yml -------------------------------------------------------------------------------- /.github/workflows/build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.github/workflows/build.yaml -------------------------------------------------------------------------------- /.github/workflows/docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.github/workflows/docs.yaml -------------------------------------------------------------------------------- /.github/workflows/publish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.github/workflows/publish.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.12 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/README.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/reference/torch_jax_interop.md: -------------------------------------------------------------------------------- 1 | ::: torch_jax_interop 2 | -------------------------------------------------------------------------------- /mkdocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/mkdocs.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/pyproject.toml -------------------------------------------------------------------------------- /torch_jax_interop/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/__init__.py -------------------------------------------------------------------------------- /torch_jax_interop/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/conftest.py -------------------------------------------------------------------------------- /torch_jax_interop/docs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/docs_test.py -------------------------------------------------------------------------------- /torch_jax_interop/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch_jax_interop/to_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_jax.py -------------------------------------------------------------------------------- /torch_jax_interop/to_jax_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_jax_module.py -------------------------------------------------------------------------------- /torch_jax_interop/to_jax_module_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_jax_module_test.py -------------------------------------------------------------------------------- /torch_jax_interop/to_jax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_jax_test.py -------------------------------------------------------------------------------- /torch_jax_interop/to_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_torch.py -------------------------------------------------------------------------------- /torch_jax_interop/to_torch_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_torch_module.py -------------------------------------------------------------------------------- /torch_jax_interop/to_torch_module_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_torch_module_test.py -------------------------------------------------------------------------------- /torch_jax_interop/to_torch_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/to_torch_test.py -------------------------------------------------------------------------------- /torch_jax_interop/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/types.py -------------------------------------------------------------------------------- /torch_jax_interop/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/torch_jax_interop/utils.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lebrice/torch_jax_interop/HEAD/uv.lock --------------------------------------------------------------------------------