Matthew Johnson 6b6789a53b

applied simple find+sed for 'master' -> 'main' (#4174) ...

* applied simple find+sed for 'master' -> 'main'

* Rename master->main in JAX API and internals (#4178)

* Started with #4174 
* Renamed Trace.master to Trace.main
* Renamed core.new_master and core.new_base_master

Co-authored-by: George Necula <gcnecula@gmail.com>

2020-08-30 11:16:51 +03:00

..

__init__.py

populating source tree

2018-11-17 18:03:33 -08:00

ad.py

applied simple find+sed for 'master' -> 'main' (#4174)

2020-08-30 11:16:51 +03:00

batching.py

applied simple find+sed for 'master' -> 'main' (#4174)

2020-08-30 11:16:51 +03:00

invertible_ad.py

Implement the invertible decorator in terms of custom_vjp (#3957)

2020-08-11 11:45:58 +02:00

masking.py

applied simple find+sed for 'master' -> 'main' (#4174)

2020-08-30 11:16:51 +03:00

partial_eval.py

applied simple find+sed for 'master' -> 'main' (#4174)

2020-08-30 11:16:51 +03:00

pxla.py

Add support for axis_index inside vmap (#4168)

2020-08-28 20:03:39 +02:00

sharded_jit.py

Use omnistaging env var even when not using absl flags for config. (#4152)

2020-08-26 14:06:27 -07:00

xla.py

Add support for axis_index inside vmap (#4168)

2020-08-28 20:03:39 +02:00