Add steps 5000 and 9999 (params + assets, train_state stripped)

280a8fe

verified

LGG100 commited on about 19 hours ago