James Bradbury 85927f01fa

Merge pull request #1456 from google/jb/no-more-dot ...

Remove `dot` primitive in favor of reusing `dot_general`

2019-10-08 17:26:17 -07:00

..

experimental

Fix typo in initial_step_size d2 calculation (#1428)

2019-10-03 23:43:34 -07:00

interpreters

fix XLA metadata for primitives with many args

2019-10-08 10:57:36 -07:00

lax

remove dot primitive in favor of dot_general

2019-10-08 14:44:10 -07:00

lib

Merge pull request #1410 from hawkinsp/master

2019-09-28 14:01:00 -04:00

nn

fixed selu

2019-10-04 22:39:35 -03:00

numpy

Revert "Revert "Add a pylintrc to make it easier to use linter (#1442)""

2019-10-08 14:39:36 -07:00

ops

Temporarily remove jit decorator on gather/scatter ops.

2019-09-16 13:57:07 -07:00

scipy

Implement jax.scipy.linalg.lu_solve.

2019-09-05 09:59:47 -04:00

tools

Remove licenses() rule comment in BUILD files.

2019-09-26 14:54:07 -07:00

__init__.py

import random in jax/__init__.py

2019-07-15 23:04:50 +01:00

abstract_arrays.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

ad_util.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

api_util.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

api.py

Add optional tuple_args argument to xla_computation.

2019-09-30 11:10:32 -07:00

BUILD

Remove licenses() rule comment in BUILD files.

2019-09-26 14:54:07 -07:00

config.py

fix config.update when absl is enabled

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

core.py

fix XLA metadata for primitives with many args

2019-10-08 10:57:36 -07:00

flatten_util.py

prevent jit from treating keyword args as static

2019-04-10 22:09:14 -07:00

lax_linalg.py

Merge pull request #1394 from j-towns/fix-scatter-caching

2019-09-26 06:48:42 -07:00

lax_reference.py

move asinh/acosh/atanh to lax_numpy.py only

2019-08-31 22:39:51 -07:00

linear_util.py

add flag for logging when jit performs compilation

2019-08-23 08:17:41 -07:00

pprint_util.py

source sync

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

random.py

Use finfo(dtype).tiny as uniform minval

2019-10-01 22:28:31 +01:00

test_util.py

test_util._cast_to_shape fix

2019-09-27 11:18:36 -07:00

tree_util.py

update jaxlib version in readme

2019-09-02 07:25:06 -07:00

util.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

version.py

bump version number for pypi

2019-09-17 08:34:01 -07:00