Jake VanderPlas
fd897745d3
Partial rollback of https://github.com/google/jax/pull/23353 as discussed in https://github.com/google/jax/pull/23353#issuecomment-2326604708
...
Reverts eed273c106af699efefc726eea1ff2b0f548f669
PiperOrigin-RevId: 670596159
2024-09-03 09:49:22 -07:00
Jake VanderPlas
7b41583414
refactor jax.lax to not depend on jax.numpy
2024-09-01 07:49:49 -07:00
Peter Hawkins
816ba91263
Use lower-case PEP 585 names for types.
...
Issue https://github.com/google/jax/issues/16537
PiperOrigin-RevId: 542969282
2023-06-23 15:12:14 -07:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
Peter Hawkins
b64e36b60f
Make QDWH-eig implementation jit-table.
...
Move QDWH-eig from jax._src.scipy.eigh to jax._src.lax.eigh, in preparation for using it to back `lax.eigh` in a future change.
PiperOrigin-RevId: 449362382
2022-05-17 18:12:48 -07:00
