Add unit tests for objax
and haiku
model params
#12
Labels
tests
Tests need to be implemented/included.
objax
and haiku
model params
#12
Currently, the unit tests of
safejax
are just covering the basic functionality and in a poor way, so those could be improved, and one of the things to improve is to extend the unit tests to make sure thatsafejax
works as expected on the serialization/deserialization of any model of the supported frameworks i.e.jax
,flax
,objax
andhaiku
The text was updated successfully, but these errors were encountered: