patrick-kidger / patrick-kidger/jaxtyping
TorchScript compatibility
Open
Nobody has claimed this yet.
feature
- Dominant language
- Python
- Stars
- 1.9k
- Forks
- 96
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
The package now does not require JAX but can also work with torch.Tensor
I was trying this, and it worked fine, but it fails when using torch.jit.script
In TorchTyping you fixed it with https://github.com/patrick-kidger/torchtyping/issues/13
Is there something similar here?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the reported failure when using torch.jit.script with torch.Tensor support, then compare the relevant behavior with the referenced TorchTyping issue 13. Done means the package remains compatible with torch.jit.script for the reported usage, with a regression test covering the case.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- devtools
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100