mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lib
History
Sergei Lebedev 2d8a20c413 Do not load Triton bindings eagerly in jax/lib/__init__.py 
Triton is only used by Pallas, so it makes sense to delay loading until Pallas
is imported.

PiperOrigin-RevId: 598131836
2024-01-13 03:01:02 -08:00
..
mlir
Use an explicit MLIR dialect registration, rather than _site_initialize_0.
2023-12-06 08:19:55 -08:00
__init__.py
Do not load Triton bindings eagerly in jax/lib/__init__.py
2024-01-13 03:01:02 -08:00
BUILD
Do not load Triton bindings eagerly in jax/lib/__init__.py
2024-01-13 03:01:02 -08:00
triton.py
Do not load Triton bindings eagerly in jax/lib/__init__.py
2024-01-13 03:01:02 -08:00