Peter Hawkins 38927153b1 Fix support for arrays with size-0 dimensions. ...

* Fix broadcasting rules to support size-0 dimensions.
* Add tests for size-0 dimensions. This required extending the test harness to support testing shapes that aren't necessarily broadcast compatible.
* Fix test utils to support size-0 dimensions.

2018-12-10 09:00:22 -05:00

..

experimental

source sync

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

interpreters

add ad_util.zeros_like translation rule

2018-12-08 09:35:24 -08:00

lib

Split out jax and jaxlib packages (#11)

2018-12-06 21:35:03 -05:00

numpy

Fix support for arrays with size-0 dimensions.

2018-12-10 09:00:22 -05:00

scipy

source sync

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

__init__.py

Set default TF log level to "1" to avoid reporting things like CPU frequency at import time. Also import jax.numpy in __init__.py because it has side effects that set up the infix operator overloading.

2018-12-05 15:55:01 -05:00

abstract_arrays.py

[JAX] Change semantics of dtype promotion to just call numpy.result_type.

2018-12-06 13:25:42 -05:00

ad_util.py

source sync

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

api_util.py

source sync

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

api.py

improve jaxlib wheel building script

2018-12-09 08:59:08 -08:00

BUILD

source sync

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

config.py

Number of test cases settable with command-line flag

2018-12-06 18:30:59 -05:00

core.py

source sync

2018-11-21 20:32:16 -08:00

lax_reference.py

source sync

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

lax.py

Fix support for arrays with size-0 dimensions.

2018-12-10 09:00:22 -05:00

linear_util.py

source sync

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

pprint_util.py

source sync

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

random.py

Split out jax and jaxlib packages (#11)

2018-12-06 21:35:03 -05:00

test_util.py

Fix support for arrays with size-0 dimensions.

2018-12-10 09:00:22 -05:00

tree_util.py

source sync

2018-11-21 20:32:33 -08:00

util.py

Added jit transformations to generated functions. Fixed bug in comparing numpy arrays for equality.

2018-12-08 00:03:34 -05:00