• Aryan's avatar
    Refactor gradient checkpointing (#10611) · c4d4ac21
    Aryan authored
    * update
    
    * remove unused fn
    
    * apply suggestions based on review
    
    * update + cleanup 🧹
    
    * more cleanup 🧹
    
    * make fix-copies
    
    * update test
    c4d4ac21
autoencoder_kl.py 24.5 KB