pytorch/functorch/examples/compilation at master · walterddr/pytorch · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
eager_fusion.py		eager_fusion.py
fuse_module.py		fuse_module.py
linear_train.py		linear_train.py
simple_function.py		simple_function.py

README.md

Compilation Examples

WARNING: Compilation is currently very experimental and example here don't work out of the box with functorch