Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
Jeppe Klitgaard
17de89b16a
feat: refactor code using pyupgrade
...
This PR upgrades legacy Python code to 3.7+ code using pyupgrade:
```sh
pyupgrade --py37-plus --keep-runtime-typing **.py
```
a
2022-05-17 22:14:05 +01:00
Peter Hawkins
2c2f4033cc
Move contents of jax.lib to jax._src.lib.
...
Add shim libraries for functions exported from jax.lib that other code seems to use in practice.
PiperOrigin-RevId: 398471863
2021-09-23 06:33:55 -07:00
Jake Vanderplas
6aa8f2461c
Fix remaining flakes and use exclude within setup.cfg ( #3371 )
2020-06-08 22:58:03 -07:00
Peter Hawkins
d59ecddfe8
Replace np -> jnp, onp -> np in examples/ ( #2971 )
...
For context, see #2370
2020-05-05 15:45:07 -04:00
Peter Hawkins
e60d5dd54c
Remove "from __future__" uses from JAX. ( #2117 )
...
The future (Python 3) has arrived; no need to request it explicitly.
2020-01-29 12:29:03 -05:00
Jonas Rauber
f8c5d98653
fixed cross-entropy losses in mnist examples ( fixes #1023 )
2019-10-27 09:43:46 +01:00
Matthew Johnson
92e5f93a29
tweak docstrings in mnist examples
2019-05-23 09:07:44 -07:00
Matthew Johnson
642d2dc802
revies optimizers api, fix misc bugs
...
* add more optimizers numerical tests
* update examples and readme with new optimziers api
* add device_values parameter to xla_call
* change optimizers.py to flatten trees and subtrees
* remove tree_map2, tree_multimap2, tree_mimomap, tree_prefixmap
* add optimizer tests: DeviceTuples and error msgs
* make the device_values arg to jit private
2019-05-03 12:44:52 -07:00
Matthew Johnson
15da530b03
add spmd mnist example
2019-03-08 09:59:03 -08:00
