Jake VanderPlas
|
f090074d86
|
Avoid 'from jax import config' imports
In some environments this appears to import the config module rather than
the config object.
|
2024-04-11 13:23:27 -07:00 |
|
Jake VanderPlas
|
fbe4f10403
|
Change to simpler import for jax.config
|
2023-04-21 11:51:22 -07:00 |
|
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
|
db2e91eba2
|
Move jax.test_util to jax._src.test_util.
Add forwarding shims for names used by external clients of JAX in practice.
PiperOrigin-RevId: 398721725
|
2021-09-24 07:02:49 -07:00 |
|
Peter Hawkins
|
3ac809ede3
|
[JAX] Move jax.util to jax._src_util.
PiperOrigin-RevId: 351234602
|
2021-01-11 14:21:07 -08:00 |
|
Matthew Johnson
|
e51163af32
|
only pass hashable values as static args
|
2020-10-16 13:11:56 -07:00 |
|
Jamie Townsend
|
7506a3e5f0
|
Fix flaky generated_fun_test.py test (#3885)
|
2020-07-28 07:27:07 -07:00 |
|
Jake VanderPlas
|
afce718eb1
|
Add ability to specify individual test targets
|
2020-06-29 11:08:57 -07:00 |
|
Peter Hawkins
|
b543652332
|
Replace np -> jnp, onp -> np in tests. (#2969)
|
2020-05-05 14:59:16 -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 |
|
George Necula
|
c1d8d3f74d
|
Add error checking that arguments of jvp are tuples
|
2019-11-27 13:12:24 +01:00 |
|
Matthew Johnson
|
d3ec0b23c5
|
fix miscellaneous bugs, incl. complex abs grad
|
2018-12-12 12:03:57 -05:00 |
|
Dougal Maclaurin
|
30124b6da1
|
Added jit transformations to generated functions. Fixed bug in comparing numpy arrays for equality.
|
2018-12-08 00:03:34 -05:00 |
|
Roy Frostig
|
e5d89ce908
|
update call to choice
|
2018-12-07 10:51:02 -08:00 |
|
Roy Frostig
|
6c2cdfc318
|
py3 compatibility: list of keys
|
2018-12-07 10:45:48 -08:00 |
|
Roy Frostig
|
fe0a26936c
|
use custom choice function in generated tests
|
2018-12-07 10:40:21 -08:00 |
|
Roy Frostig
|
479f256abd
|
py3 iterator compatibility
|
2018-12-07 10:22:15 -08:00 |
|
Roy Frostig
|
6fb8a8bc89
|
use print as function
|
2018-12-07 09:43:28 -08:00 |
|
Dougal Maclaurin
|
8b88027df0
|
Number of test cases settable with command-line flag
|
2018-12-06 18:30:59 -05:00 |
|
Dougal Maclaurin
|
ebc6cd1e03
|
Step through sizes in test case generation
|
2018-12-06 18:02:43 -05:00 |
|
Dougal Maclaurin
|
3dbf41f3e6
|
Generated function tests working with bazel
|
2018-12-06 17:31:52 -05:00 |
|
Dougal Maclaurin
|
c3374a9d5f
|
Added build rule for generated_fun_test (formerly quickish_check)
|
2018-12-06 17:04:00 -05:00 |
|