jax
072e6f78 - replace PRNGKey class with uint32[2] array

Commit

7 years ago

replace PRNGKey class with uint32[2] array

References

#179 - add support for vmap over PRNGKeys

Author

mattjj

mattjj

Parents

Loading