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
|
160dfd343a
|
Revert import path changes to examples/ and benchmarks/
PiperOrigin-RevId: 352911869
|
2021-01-20 17:35:55 -08:00 |
|
Peter Hawkins
|
929a684a39
|
Small cleanups to dependency structure.
PiperOrigin-RevId: 352853244
|
2021-01-20 12:43:28 -08: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
|
15da530b03
|
add spmd mnist example
|
2019-03-08 09:59:03 -08:00 |
|
Peter Hawkins
|
95483c76e9
|
Expose logsumexp as scipy.special.logsumexp.
scipy.misc.logsumexp is deprecated and appears slated to be removed entirely in scipy 1.3.
|
2019-02-24 11:49:15 -05:00 |
|
Matthew Johnson
|
c54e82ceff
|
be more explicit about last layer in example
closes #119
|
2018-12-16 07:23:13 -08:00 |
|
Matthew Johnson
|
e682af7ea5
|
increase mnist example batch size to 128
|
2018-12-13 18:24:35 -08:00 |
|
Roy Frostig
|
bec24999a8
|
import more examples in examples_test, fix resulting errors
|
2018-12-12 16:09:20 -08:00 |
|
Dougal Maclaurin
|
2df36f7510
|
Made a shim to handle configuration without having absl parse command-line flags.
PiperOrigin-RevId: 223391288
|
2018-11-29 13:44:54 -08:00 |
|
Matthew Johnson
|
d318c68827
|
source sync
PiperOrigin-RevId: 222822439
|
2018-11-27 16:51:11 -08:00 |
|
Matthew Johnson
|
dbf3c606f5
|
source sync
PiperOrigin-RevId: 222500675
|
2018-11-27 16:51:00 -08:00 |
|
Roy Frostig
|
7a313bf622
|
source sync fixups
|
2018-11-21 20:32:33 -08:00 |
|
Matthew Johnson
|
323be694a7
|
source sync
PiperOrigin-RevId: 222448341
|
2018-11-21 20:22:47 -08:00 |
|
Roy Frostig
|
a3619ca89d
|
source sync
PiperOrigin-RevId: 222153576
|
2018-11-21 20:22:30 -08:00 |
|
Matthew Johnson
|
51fc713089
|
remove absl from examples, fix import statements
|
2018-11-21 12:10:31 -08:00 |
|
Matthew Johnson
|
46c6a9170f
|
sync updates
|
2018-11-19 07:47:59 -08:00 |
|