Automatic activation checkpointing and tensor swapping for DNN training using torch.fx graph profiling.
-
Updated
Aug 30, 2026 - Jupyter Notebook
#
torch-fx
Here are 2 public repositories matching this topic...
From-scratch graph optimizer for torch.compile: constant folding, CSE and DCE on PyTorch FX graphs.
-
Updated
Oct 15, 2026 - Python
Add this topic to your repo
To associate your repository with the torch-fx topic, visit your repo's landing page and select "manage topics."