torch_cudagraph_fx is a make_fx-style tracer and interpreter oriented to graph capture for CUDA-graphable training steps. It is a prototype of the design in DESIGN.md. Your code must be CUDA-graphable ...
This section assumes you have already created your first driver and started working on your implementation of the ResourceDriverHandlerCapability. It's at this stage ...