[Question] composable activation checkpoint

[Irvingwangjr]

I'm looking for a way that not to use module wrapper to apply activation checkpoint, and I found this https://github.com/pytorch/pytorch/pull/87664/files.
Is this method works fine? Or it just a demo code

[tianyu-l]

cc @soulitzer