public readonly property checkpointStates as ImmutableList<AdamState<T>>
Snapshot of parameter states for checkpoint transport, not tensor copies.
print(optimizer.checkpointStates.size)