Jake Vanderplas 0db57cb541

Fix validation code in lax.conv (#3279)

2020-06-03 10:33:19 -07:00

..

experimental

keep old name 'packed_state' of OptimizerState

2020-06-03 07:32:44 -07:00

interpreters

Fix pytype errors. (#3291)

2020-06-02 10:26:43 -04:00

lax

Fix validation code in lax.conv (#3279)

2020-06-03 10:33:19 -07:00

lib

Add a JAX flag to avoid most optimizations. (#3208)

2020-05-26 15:21:22 -04:00

nn

remove some trailing whitespace (#3287)

2020-06-02 17:37:20 -07:00

numpy

Add np.polyadd (#3261)

2020-06-03 10:26:35 -07:00

ops

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

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

scipy

remove some trailing whitespace (#3287)

2020-06-02 17:37:20 -07:00

third_party

remove some trailing whitespace (#3287)

2020-06-02 17:37:20 -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

Exported lax from jax/__init__.py (#3135)

2020-05-19 15:40:03 -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

Add support for buffer donation in jit and pmap. (#2936)

2020-05-31 15:00:16 -07:00

api.py

make pmap axis checking an exception, hoist (#3239)

2020-06-02 20:28:59 -07:00

BUILD

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

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

config.py

Test code in docs and api.py docstrings (#2994)

2020-05-16 16:19:24 +03:00

core.py

moved check_jaxpr code around to match eval_jaxpr (#3240)

2020-06-02 19:10:55 -07:00

custom_derivatives.py

Fix sign error in custom_jvp / custom_vjp. (#3213) (#3219)

2020-05-28 10:21:39 -07: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

Expand type support for random uniform() & randint() (#3138)

2020-05-19 14:19:00 -07: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

Make check_dtypes, atol, and rtol keyword-only arguments in jax.test_… (#3280)

2020-06-01 17:19:23 -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

remove some trailing whitespace (#3287)

2020-06-02 17:37:20 -07:00

test_util.py

Fix bug where jnp.array returned a classic NumPy array, sometimes wit… (#3283)

2020-06-01 19:29:26 -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

merge constvars when forming cond branch jaxprs

2020-05-13 21:14:41 -07:00

version.py

update version and changelog for pypi

2020-06-03 07:45:53 -07:00