Matthew Johnson a17f8e4ca8 add jaxpr eqn structured input, transpose progress ...

Co-authored-by: Dougal Maclaurin <dougalm@google.com>

2019-05-08 13:41:19 -07:00

..

experimental

back to original tree_mimomap optimizers solution

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

interpreters

add jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -07:00

lib

tweak to fix xla_bridge.device_count behavior

2019-03-20 16:29:43 -07:00

numpy

Merge pull request #528 from kroq-gar78/cross

2019-03-26 07:05:35 -07:00

ops

Add missing jax/ops/ to jax/BUILD.

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

scipy

no need to import onp

2019-03-14 23:13:30 -04: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 jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -07:00

ad_util.py

add jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -07:00

api_util.py

add partial value lattice join, cond support

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

api.py

Merge pull request #525 from google/jarrett-jvps-2

2019-03-25 18:37:05 -07:00

BUILD

Add missing jax/ops/ to jax/BUILD.

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

config.py

fix failing tests (misc small bugs)

2018-12-13 11:52:41 -08:00

core.py

add jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -07:00

flatten_util.py

generalize jacfwd and jacrev to handle pytrees

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

initial_style.py

add jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -07:00

lax_linalg.py

Add batching rule for triangular solve.

2019-03-10 17:31:51 -04:00

lax_reference.py

source sync

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

lax.py

add jaxpr eqn structured input, transpose progress

2019-05-08 13:41:19 -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

fix bernoulli shape bug

2019-03-02 19:11:16 -05:00

scan.py

victory! patial eval of scan (+ linearize!)

2019-05-08 13:41:15 -07: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-02 11:02:18 -08:00