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/lax
History
jax authors 3e3542b0d6 Upgrade logistic (sigmoid) function into a lax primitive. 
This allows us to lower it to `mhlo.logistic`, which allows XLA to generate more efficient code.

PiperOrigin-RevId: 469841487
2022-08-24 15:39:37 -07:00
..
__init__.py
Upgrade logistic (sigmoid) function into a lax primitive.
2022-08-24 15:39:37 -07:00
linalg.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00