Add topic tags

[Beliavsky]

I suggest adding the topics cuda, checkpoint in the About section, as explained at https://docs.github.com/en/repositories/managing-your-repositorys-settings-and-features/customizing-your-repository/classifying-your-repository-with-topics

[jesus-ramos]

Thanks, I've added those.