Fix TransformerEngine and Activation checkpointing interaction for thunderFX#1473

Draft

kshitij12345 wants to merge 5 commits intoLightning-AI:mainfrom kshitij12345:te-activation-ckpt

+70-16

Commits on Nov 26, 2024

fix te and activation checkpointing for thunderFX
kshitij12345
committed

Commits on Nov 27, 2024

Merge branch 'main' into te-activation-ckpt
tfogal
authored

Commits on Dec 30, 2024