mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 17:36:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/pallas
History
jax authors 4991aac30f Import new version of Triton 
PiperOrigin-RevId: 590496513
2023-12-13 01:24:24 -08:00
..
mosaic
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00
triton
Import new version of Triton
2023-12-13 01:24:24 -08:00
__init__.py
[Pallas] Upstream pallas to JAX
2023-08-01 16:43:13 -07:00
BUILD
Load Python rules from rules_python
2023-08-24 10:22:57 -07:00
core.py
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00
indexing.py
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00
pallas_call.py
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00
primitives.py
[pallas] dot fixes.
2023-10-16 12:35:43 -07:00
utils.py
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00