This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-17 20:36:05 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
pallas
/
triton
History
jax authors
2c165bffc9
[pallas:triton] Lift dot_general restriction on minimal tile size for a.
...
PiperOrigin-RevId: 725605869
2025-02-11 06:27:16 -08:00
..
__init__.py
Removed the double re-exporting of Pallas GPU/TPU APIs
2024-06-10 05:59:09 -07:00
BUILD
jax.core: deprecate a number of APIs
2024-12-10 11:11:32 -08:00
core.py
[Pallas] Fix type annotation on TritonCompilerParams
2024-12-04 12:47:41 -08:00
lowering.py
[pallas:triton] Lift dot_general restriction on minimal tile size for a.
2025-02-11 06:27:16 -08:00
pallas_call_registration.py
[pallas:triton] Really revert to the lowering using Triton IR
2025-02-07 06:55:14 -08:00
primitives.py
jax.core: deprecate a number of APIs
2024-12-10 11:11:32 -08:00
