joao guilherme d2f84d635b

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

..

experimental

Fix error message in optimizers piecewise_constant. (#3061)

2020-05-12 11:09:02 -04:00

interpreters

Refactor aval_to_result_handler to take unsharded aval. (#3067)

2020-05-12 14:34:31 -07:00

lax

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

lib

Import tpu_driver after xla_client (#3064)

2020-05-12 11:05:03 -07:00

nn

Uses jnp.square instead of power. (#3036)

2020-05-12 11:04:53 -04:00

numpy

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

ops

Update internal aliases to lax_numpy to jnp instead of np. (#2975)

2020-05-05 20:41:57 -04:00

scipy

Uses jnp.square instead of power. (#3036)

2020-05-12 11:04:53 -04:00

third_party

Cleanup: move _wraps into jax.numpy._utils. (#2987)

2020-05-06 15:17:55 -07:00

tools

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

__init__.py

Remove jax.np from the jax namespace (use jax.numpy instead). (#3010)

2020-05-08 10:04:19 -04:00

abstract_arrays.py

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

ad_util.py

stop_gradient_p -> ad_util.py, re-enable some mypy (#2806)

2020-04-23 13:12:24 -07:00

api_util.py

Move _flatten_axes to api_util.py (#3041)

2020-05-11 11:04:57 -07:00

api.py

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

BUILD

Implementation of id_tap/id_print using outfeed. (#3006)

2020-05-08 17:18:11 +03:00

config.py

Add flag to enable checking, and turn on checking in tests. (#2900)

2020-05-01 09:16:31 +03:00

core.py

Implementation of id_tap/id_print using outfeed. (#3006)

2020-05-08 17:18:11 +03:00

custom_derivatives.py

Replace np -> jnp, onp -> np in more places. (#2973)

2020-05-05 16:40:41 -04:00

dlpack.py

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

dtypes.py

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

flatten_util.py

Replace np -> jnp, onp -> np in more places. (#2973)

2020-05-05 16:40:41 -04:00

lax_linalg.py

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

lax_reference.py

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

lazy.py

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

linear_util.py

revamp custom_jvp/vjp implementation to fix bugs

2020-03-29 19:35:01 -07:00

pprint_util.py

Remove "from __future__" uses from JAX. (#2117)

2020-01-29 12:29:03 -05:00

profiler.py

Add a jax.profiler module that exposes the new TensorFlow profiler in… (#2354)

2020-03-05 12:07:57 -05:00

random.py

Avoid recompilation of rolled loops in threefry2x32. (#3069)

2020-05-12 15:03:22 -07:00

test_util.py

Change instances of onp to np and np to jnp (#3044)

2020-05-12 20:37:05 -04:00

tree_util.py

Expose functools.reduce initializer argument to tree_util.tree_reduce (#2935)

2020-05-05 11:11:10 +03:00

util.py

Removed unused imports (#2385)

2020-03-09 20:42:08 +01:00

version.py

Update jax version to 0.1.67 (#3065)

2020-05-12 11:17:22 -07:00