8000 GitHub - nunoplopes/torchy: A tracing JIT compiler for PyTorch
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
This repository was archived by the owner on Feb 7, 2024. It is now read-only.

nunoplopes/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 compiler for PyTorch

Topics

Resources

License

Stars

Watchers

Forks

Languages

0