mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 22:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 508 Branches 263 Tags
rocm-main
Commit Graph

24 Commits

Author SHA1 Message Date
Peter Hawkins
8ab33669e2 Add a variant of safe_map() that has no return value, named foreach(). 
This avoids a bunch of list bookkeeping in cases where we are iterating only for a side effect and do not care about the results.

I would have named this iter() to match OCaml's list.iter(), but unfortunately iter() is a Python builtin.

PiperOrigin-RevId: 736859418
 2025-03-14 07:42:48 -07:00
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 () 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. () 2020-05-05 14:59:16 -04:00
Peter Hawkins
e60d5dd54c
Remove "from __future__" uses from JAX. () 
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