v0.5.1

NTT123 released this 23 Nov 06:42

· 40 commits to main since this release

ae1718b

Add state_dict and load_state_dict methods.
Add pax.purecall function. Prefer this to the pax.module_and_value function.

Assets 2