Neil Kichler issues

Repositories
Issues
Comments

Results 2 issues of


                                            Neil Kichler

Support Jax version 4.25.0 and higher.

Currently, the basic example from the README.md does not run for versions of JAX>=4.25.0. This is due to the removal of the count attribute in the jax.core.Var class and its...

Fix type hints in taylor_bounds to support int and float arguments.

In the README example, Pyright complains about the arguments to taylor_bounds since the types of the Callable are restricted to jnp.ndarray and not types.NDArraylike which also supports int, float etc.