Sharad Vikram af2306c0a8 Refactor effects system to use effect types, not objects

2023-02-17 17:40:08 -08:00

..

control_flow

Refactor effects system to use effect types, not objects

2023-02-17 17:40:08 -08:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ann.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00

convolution.py

Implement lowering for convolutions with dynamic padding

2023-02-14 00:55:45 -08:00

eigh.py

Replace uses of jnp.ndarray with jax.Array inside JAX.

2023-02-15 14:53:00 -08:00

fft.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00

lax.py

Refactor effects system to use effect types, not objects

2023-02-17 17:40:08 -08:00

linalg.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00

other.py

Replace uses of jnp.ndarray with jax.Array inside JAX.

2023-02-15 14:53:00 -08:00

parallel.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00

qdwh.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

slicing.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00

stack.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

svd.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

utils.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

windowed_reductions.py

migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching

2023-02-09 15:11:57 -08:00