mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 15:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
12,176 Commits 508 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Jake VanderPlas
12900bf6ab [x64] make jnp.mgrid compatible with strict dtype promotion 2022-06-01 10:21:49 -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
YouJiacheng
f0b8d70d15 simplify _IndexGrid 2022-04-15 04:17:27 +08:00
jax authors
d7bd93f6d1 Merge pull request #9925 from jakevdp:refactor-index-ops 
PiperOrigin-RevId: 435408095
 2022-03-17 11:56:38 -07:00
Jake VanderPlas
0a72adbd5e lax_numpy: factor out indexing tricks 2022-03-17 11:05:45 -07:00