Roy Frostig 4c73b18b53 Merge branch 'master' into parallelize

2019-04-09 10:47:18 -07:00

..

experimental

Fix stax x64 dtype issues

2019-04-03 15:03:29 +01:00

interpreters

Merge branch 'master' into parallelize

2019-04-03 15:32:06 -07:00

lib

Change jaxlib version test to import jaxlib.version first.

2019-04-01 08:36:35 -07:00

numpy

Fixed numpy.zeros shape generator error

2019-04-08 19:15:47 +02:00

ops

DOC: Fix typo in ops.index_update

2019-03-30 17:18:53 +09:00

scipy

Merge pull request #548 from fehiepsi/pareto

2019-03-31 06:15:45 -07:00

__init__.py

Use a regular import to add jax.__version__ rather than exec() trickery.

2019-02-19 11:38:28 -05:00

abstract_arrays.py

Add np.longlong to the set of JAX array types.

2019-04-01 15:49:12 -07:00

ad_util.py

peval.py works again (some paired w/ @dougalm)

2019-02-22 07:53:28 -08:00

api_util.py

add partial value lattice join, cond support

2019-03-02 17:37:38 -08:00

api.py

fix typo in xla_computation

2019-04-04 17:40:48 -07:00

BUILD

Add missing jax/ops/ to jax/BUILD.

2019-03-06 09:07:15 -05:00

config.py

correctly update jax config.values after absl flag parsing

2019-04-04 02:09:35 -07:00

core.py

add __bool__ to jaxtuples / abstracttuples

2019-03-02 21:43:40 -08:00

flatten_util.py

generalize jacfwd and jacrev to handle pytrees

2019-01-06 12:49:41 -08:00

lax_linalg.py

Refactor handling of XLA backends.

2019-03-29 11:09:56 -04:00

lax_reference.py

source sync

2018-11-21 20:22:51 -08:00

lax.py

Merge branch 'master' into parallelize

2019-04-09 10:47:18 -07:00

linear_util.py

Minor typo fixes.

2019-03-12 15:28:07 -04:00

pprint_util.py

source sync

2018-11-21 20:22:54 -08:00

random.py

Rm unnecessary dtype arg name

2019-04-02 10:56:44 +01:00

test_util.py

reconcile _CheckAgainstNumpy arg order common use

2019-03-22 17:09:35 -07:00

tree_util.py

comment-out new pytree check with a TODO

2019-03-22 07:38:40 -07:00

util.py

fix pmap performance bug, dont always copy to host

2019-03-26 08:35:34 -07:00

version.py

bump version for pypi

2019-04-07 17:50:43 -07:00