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/nn
History
Peter Hawkins 29d03160e3 Remove _ prefix from functions in jax._src.dtypes. 
to_inexact_dtype and to_complex_dtype are used across the JAX code base,
so they shouldn't have _ prefixes.
2022-08-12 12:51:09 +00:00
..
__init__.py
Move jax.nn implementation into jax._src.nn.
2020-10-17 13:45:01 -04:00
functions.py
Remove _ prefix from functions in jax._src.dtypes.
2022-08-12 12:51:09 +00:00
initializers.py
Remove _ prefix from functions in jax._src.dtypes.
2022-08-12 12:51:09 +00:00