mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Import jax.nn.functions by default to fix breakage.

Browse Source
This commit is contained in:
Peter Hawkins 2020-10-17 14:51:39 -04:00
parent 3ddd3905a4
commit b07848359b
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/nn/__init__.py
View File

@ -15,6 +15,10 @@
"""Common functions for neural network libraries."""
# flake8: noqa: F401
# TODO(phawkins): remove this import after fixing callers
from . import functions
from . import initializers
from jax._src.nn.functions import (
celu,