LICENSE
README.md
pyproject.toml
torch_remat/__init__.py
torch_remat/_annotate.py
torch_remat/_api.py
torch_remat/_compat.py
torch_remat/_oom.py
torch_remat/_placeholder.py
torch_remat/_pytree.py
torch_remat/_recompute_boundary.py
torch_remat/_region.py
torch_remat/_reporting.py
torch_remat/_saved_report.py
torch_remat/_trace.py
torch_remat/_types.py
torch_remat/_view.py
torch_remat.egg-info/PKG-INFO
torch_remat.egg-info/SOURCES.txt
torch_remat.egg-info/dependency_links.txt
torch_remat.egg-info/requires.txt
torch_remat.egg-info/top_level.txt