Lena Martens ecad419cf3 Support grad with integer arguments. ...

- Add float0 and set-up at_least_vspace to return float0
values for int/bool primals
- Use Zero to wrap float0 tangents so they're correctly ignored in jvp
rules
- Add float0 handlers to XLA to support jit
- Fix convert_element_type and tie_in jvp rules

2020-09-28 19:07:04 +01:00

..

__init__.py

Implement jnp.array_equiv

2020-09-21 10:37:55 -07:00

_util.py

Add test of jax.numpy vs numpy call signatures (#3628)

2020-07-01 16:52:41 -07:00

fft.py

Fix code quality issues (#4302)

2020-09-17 09:21:18 -07:00

lax_numpy.py

Support grad with integer arguments.

2020-09-28 19:07:04 +01:00

linalg.py

Modifies eig_p and related operations to take advantage of the new jaxlib geev API (#4266)

2020-09-15 11:45:15 +03:00

polynomial.py

Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)

2020-07-09 16:31:08 -07:00

vectorize.py

Edit documentation. (#3358)

2020-06-08 10:37:50 -04:00