https://github.com/nunoplopes/torchy
A tracing JIT compiler for PyTorch
https://github.com/nunoplopes/torchy
jit-compiler lazy-evaluation pytorch tracing-jit
Last synced: 8 months ago
JSON representation
A tracing JIT compiler for PyTorch
- Host: GitHub
- URL: https://github.com/nunoplopes/torchy
- Owner: nunoplopes
- License: mit
- Archived: true
- Created: 2021-03-25T20:15:03.000Z (about 5 years ago)
- Default Branch: main
- Last Pushed: 2021-12-11T12:58:49.000Z (over 4 years ago)
- Last Synced: 2025-01-28T17:38:37.741Z (over 1 year ago)
- Topics: jit-compiler, lazy-evaluation, pytorch, tracing-jit
- Language: C++
- Homepage:
- Size: 5.27 MB
- Stars: 12
- Watchers: 2
- Forks: 0
- Open Issues: 10
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Torchy
======
A tracing JIT for PyTorch.
WIP; don't use.
Install
-------
```
$ python setup.py install
```
Run
---
Torchy shouldn't require any change beyond adding a call to `torchy.enable()`.
Example:
```python
import torch
import torchy
torchy.enable()
x = torch.tensor(((3.,2.), (4.,5.)))
y = torch.tensor(((5.,6.), (7.,1.)))
w = x.add(x)
z = x.add(y)
w = None # w not computed
print(z)
```