Matthew Johnson 4128240964 update version and changelog for pypi

2020-10-13 23:01:30 -07:00

..

experimental

[jax2tf] Added conversion paths without einsum for dot_general.

2020-10-13 18:13:34 +02:00

image

Make scale_and_translate take spatial dimensions

2020-09-21 16:20:17 -07:00

interpreters

Merge pull request #4161 from jakevdp:raise-to-shaped

2020-10-08 13:15:48 -07:00

lax

Merge pull request #4530 from jakevdp:cleanup

2020-10-13 09:45:13 -07:00

lib

Remove scalar float0 XLA representation.

2020-10-01 18:02:31 +00:00

nn

Adds an approximate=... keyword argument to jax.nn.gelu to select between the approximate and exact formulations of gelu.

2020-10-02 09:48:07 -04:00

numpy

Merge pull request #4556 from jakevdp:index-fix

2020-10-13 10:09:06 -07:00

ops

Add ceil_of_ratio util and bucket_size TODO.

2020-09-21 14:45:37 -07:00

scipy

Use jvp(expm) to compute expm_frechet.

2020-09-17 21:25:40 -07:00

third_party

Cleanup: avoid jnp.prod & np.prod on array shapes (#4086)

2020-08-18 10:17:38 -07:00

tools

Fix code quality issues (#4302)

2020-09-17 09:21:18 -07:00

__init__.py

Support grad with integer arguments.

2020-09-28 19:07:04 +01:00

abstract_arrays.py

Support grad with integer arguments.

2020-09-28 19:07:04 +01:00

ad_util.py

Support grad with integer arguments.

2020-09-28 19:07:04 +01:00

api_util.py

fix typo

2020-09-18 19:41:59 -07:00

api.py

Improve pmap axis error in the presence of pytrees

2020-10-13 11:22:59 -07:00

BUILD

Internal change

2020-10-13 14:36:45 -07:00

config.py

omnistaging on by default (#4038)

2020-09-15 08:06:46 -07:00

core.py

Enrich the error messages with the bound names that are available.

2020-10-12 20:33:27 +02:00

custom_derivatives.py

raise_to_shaped: preserve weak_type by default

2020-10-08 11:53:52 -07:00

dlpack.py

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)

2020-05-11 17:43:55 -04:00

dtypes.py

Support grad with integer arguments.

2020-09-28 19:07:04 +01:00

flatten_util.py

deflake

2020-06-12 15:41:49 -07:00

jaxpr_util.py

move experimental.jaxpr_stats to jaxpr_util

2020-08-18 18:07:38 -07:00

lax_linalg.py

Enable fast TPU LU decomposition for complex types.

2020-09-22 22:03:02 -04:00

lax_reference.py

refine population_count type check (#3887)

2020-07-28 19:46:00 -07:00

lazy.py

deflake remainder of jax (#3343)

2020-06-06 10:51:34 -07:00

linear_util.py

Merge branch 'master' into debug-nans-no-store-error

2020-10-08 12:28:11 -07:00

pprint_util.py

Attach source info to Jaxpr equations. (#3421)

2020-06-17 16:35:36 -07:00

profiler.py

Add a jax.profiler.StepTraceContext API. (#3591)

2020-07-10 17:20:06 -07:00

random.py

Add stop_gradients around lax.nextafter to fix TFP gradient errors for jax.random.truncated_normal.

2020-10-13 09:16:29 -04:00

source_info_util.py

Attach source info to Jaxpr equations. (#3421)

2020-06-17 16:35:36 -07:00

test_util.py

Merge pull request #4419 from rsepassi:master

2020-10-01 10:39:51 -07:00

traceback_util.py

early return for non-existent path prefixes

2020-10-07 11:59:12 -07:00

tree_util.py

Compare treedefs by num_leaves not traversal_ in tree_transpose. (#3659)

2020-07-30 13:31:17 -04:00

util.py

Merge pull request #4299 from zhangqiaorjc:segsum

2020-09-21 16:48:00 -07:00

version.py

update version and changelog for pypi

2020-10-13 23:01:30 -07:00