Skip to content

A tracing JIT for PyTorch

License

Notifications You must be signed in to change notification settings

microsoft/torchy

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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:

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)

About

A tracing JIT for PyTorch

Resources

License

Code of conduct

Security policy

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages