Raghav Kandmergify[bot] e2252c0a5e [perf] Mark LayerwiseOffloadHook entry points torch.compiler.disable (remove per-layer graph break) (#1365) ...

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-05-20 03:09:23 +00:00

..

__init__.py

[feat] Hooks API and layerwise offloading for all DiTs (#1006)

2026-01-17 11:22:02 -08:00

activation_trace.py

[infra]: Add activation trace hooks for pipeline debugging (#1293)

2026-05-07 16:38:13 -07:00

hooks.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

layerwise_offload.py

[perf] Mark LayerwiseOffloadHook entry points torch.compiler.disable (remove per-layer graph break) (#1365)

2026-05-20 03:09:23 +00:00