Ayaka
|
859eacb5a1
|
Fix mypy error
|
2024-08-27 16:57:53 +01:00 |
|
Ruturaj4
|
e06be544d4
|
[ROCm] improve gpu script
|
2024-08-19 15:13:58 -05:00 |
|
Yue Sheng
|
eb571c984a
|
Fix lint in run_single_gpu.py
PiperOrigin-RevId: 658933291
|
2024-08-02 16:03:03 -07:00 |
|
Rahul Batra
|
7d6fa3c05b
|
[ROCm]: Add support to continue on fail, fix script paths and update Dockerfile to add necessary packages
|
2024-08-01 17:55:15 -05:00 |
|
Peter Hawkins
|
7f4ef63cd8
|
Run pyupgrade --py310-plus.
Also apply manual fixes to import sorting and unused imports.
|
2024-06-26 16:10:18 -04:00 |
|
Dan Foreman-Mackey
|
1c640b25cb
|
Fix pre-commit formatting errors
It looks like #21916 introduced some whitespace issues that are causing
our pre-commits to fail.
|
2024-06-26 06:38:51 -04:00 |
|
zahiqbal
|
ef7694f26a
|
[ROCM]: Generating pytest html logs from unit-tests.
|
2024-01-24 15:08:35 +00:00 |
|
Rohit Santhanam
|
b815ac9d8e
|
[ROCm] Upgrade to ROCm 5.3 and associated enhancements
|
2022-10-01 04:45:26 -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 |
|
jax authors
|
cf9a900d78
|
Merge pull request #9584 from ROCmSoftwarePlatform:rocm_refactor_jaxlib
PiperOrigin-RevId: 432236852
|
2022-03-03 11:11:02 -08:00 |
|
Reza Rahimi
|
a0d9d81f92
|
Update JAX to use new math libraries in ROCm-5.0.
|
2022-03-01 20:02:15 +00:00 |
|