flax
Add static_argnums to nn.checkpoint
#2457
Merged

Loading