use intermediate activation checkpointing in pytorch to trade compute for increased batch size
use intermediate activation checkpointing in pytorch to trade compute for increased batch size