[JAX] Replace jax.numpy.NINF with -jax.numpy.inf.

google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

Apache License 2.0

1.43k stars 202 forks source link

[JAX] Replace jax.numpy.NINF with -jax.numpy.inf. #1308

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

[JAX] Replace jax.numpy.NINF with -jax.numpy.inf.

The aliases INF and NINF are slated for removal from upstream NumPy in version 2.0. jax.numpy mirrors upstream NumPy, so we will eventually also remove them from jax.numpy as well.