Add inference checkpoint (optimizer/cache stripped)

9411949

verified

matthewh6 commited on 21 days ago