v0.2.4
fixed bug in activation checkpointing test (#387)
[doc] update rst and docstring (#1351)
html refactor (#555)