Junda Su e3b88bb12a [feat] Add a typed environment-variable registry, policy doc, and contract test (#1896)

2026-09-29 12:50:45 -07:00

..

__init__.py

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

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

activation_trace.py

[feat] Add a typed environment-variable registry, policy doc, and contract test (#1896)

2026-09-29 12:50:45 -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