Anselm Levskaya 116e329e10 correctly update jax config.values after absl flag parsing

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

..

experimental

back to original tree_mimomap optimizers solution

2019-03-22 07:12:25 -07:00

interpreters

make DeviceArray.__repr__ show it's not an ndarray

2019-04-03 09:43:44 -07:00

lib

Change jaxlib version test to import jaxlib.version first.

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

numpy

Refactor handling of XLA backends.

2019-03-29 11:09:56 -04: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

graphviz: concat strings only at the end

2019-04-02 21:18:20 -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 pull request #570 from hawkinsp/master

2019-04-03 13:15:47 -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 number for pypi

2019-03-28 06:42:04 -07:00