The "Ideal" PyTorch FLOP Counter (with __torch_dispatch__)

This is an updated version of the FLOP counter that uses TorchDispatchMode instead

1 Like