Add inference checkpoint (optimizer/cache stripped)

9411949

verified

matthewh6 commited on 18 days ago