Update libs, add **predict_kwargs

asmith26 / jax_toolkit

A collection of jax functions to help with common machine/deep learning related functionality.

https://asmith26.github.io/jax_toolkit/

Apache License 2.0

5 stars 0 forks source link

Closed asmith26 closed 3 years ago