feat: updated activation checkpointing

#14

Untested so far

Markus28 changed pull request status to open
Markus28 changed pull request status to merged

Sign up or log in to comment