gradient checkpoint feat was introduced here https://github.com/keras-team/keras/pull/20743 but currently there is no user guide how to adop it or use it.