mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 13:06:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/pallas
History
Richard Levasseur f891cbf64b Load Python rules from rules_python 
PiperOrigin-RevId: 559789250
2023-08-24 10:22:57 -07:00
..
mosaic
Load Python rules from rules_python
2023-08-24 10:22:57 -07:00
triton
Load Python rules from rules_python
2023-08-24 10:22:57 -07: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
[Pallas] Upstream pallas to JAX
2023-08-01 16:43:13 -07:00
indexing.py
[Pallas] Upstream pallas to JAX
2023-08-01 16:43:13 -07:00
pallas_call.py
[pallas] Minor cleanup to pallas_call_p JVP code.
2023-08-11 02:30:35 -07:00
primitives.py
[Pallas] Upstream pallas to JAX
2023-08-01 16:43:13 -07:00
utils.py
[Pallas] Add support for lax.fori_loop to Mosaic lowering
2023-08-04 13:43:43 -07:00