Peter Hawkins 0b1d2fc3d1

Avoid accidental type promotion in gamma sampler gradient. (#2150) ...

Reformat gamma sampler to use 2 space indent, consistent with the rest of JAX.

2020-02-03 12:44:46 -05:00

..

experimental

Make stax pooling layers accept spec=None (#2145)

2020-02-03 10:31:12 -05:00

interpreters

fix remat with nontrivial env (#2136)

2020-01-31 23:47:30 -08:00

lax

VJP of cond, via partial eval + transpose (#2091)

2020-01-30 15:03:00 -08:00

lib

Add num_partitions parameter to get_compile_options. (#2052)

2020-01-29 11:35:48 -08:00

nn

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

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

numpy

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

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

ops

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

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

scipy

Implement scipy.stats.logistic (#1993)

2020-01-30 20:19:01 -05:00

tools

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

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

__init__.py

Import nn in jax/__init__.py.

2019-11-01 09:28:48 +00:00

abstract_arrays.py

Ensure ShapedArray.shape is always a tuple of builtins integers (#2039)

2020-01-29 14:24:11 -08:00

ad_util.py

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

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

api_util.py

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

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

api.py

fix remat with nontrivial env (#2136)

2020-01-31 23:47:30 -08:00

BUILD

Minimal changes to make Jax pass a pytype check. (#2024)

2020-01-18 08:26:23 -05:00

config.py

fix config.update when absl is enabled

2019-10-02 16:08:05 -07:00

core.py

Rename Tracer.trace to Tracer._trace. (#2114)

2020-01-29 16:23:27 -05:00

dlpack.py

Initial implementation of DLPack support. (#2123)

2020-01-29 23:19:14 -05:00

dtypes.py

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

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

flatten_util.py

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

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

lax_linalg.py

Better batching rule for triangular_solve (#2138)

2020-02-03 09:27:03 -08:00

lax_reference.py

Fix test failures in incomplete gamma functions with Jaxlib 0.1.38. (#2118)

2020-01-29 13:24:45 -05:00

lazy.py

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

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

linear_util.py

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

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

pprint_util.py

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

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

random.py

Avoid accidental type promotion in gamma sampler gradient. (#2150)

2020-02-03 12:44:46 -05:00

test_util.py

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

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

tree_util.py

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

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

util.py

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

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

version.py

Bump JAX version to 0.1.58.

2020-01-28 09:24:26 -05:00