search

ai-computing / aicomp

Other
6 stars 0 forks source link

Activation checkpoint support #10

Open ememos opened 3 months ago

ememos commented 3 months ago

Activation checkpoint is needed

ememos commented 3 months ago

produce_forward_output function added.