Ë
    ñukiö  ã                   ó°   — d Z ddlmZ ddlmZ ddlmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' y)z.Common functions for neural network libraries.é    )Útanh)Úinitializers)"ÚceluÚeluÚgeluÚgluÚhard_sigmoidÚ	hard_siluÚ
hard_swishÚ	hard_tanhÚ
leaky_reluÚlog_sigmoidÚlog_softmaxÚ
logmeanexpÚ	logsumexpÚstandardizeÚone_hotÚreluÚidentityÚrelu6Údot_product_attentionÚget_scaled_dot_general_configÚscaled_dot_generalÚscaled_matmulÚseluÚsigmoidÚ	soft_signÚsoftmaxÚsoftplusÚsparse_plusÚsparse_sigmoidÚsiluÚswishÚ
squareplusÚmishÚlog1mexpN)(Ú__doc__Ú	jax.numpyr   Újax.nnr   Újax._src.nn.functionsr   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   © ó    úJ/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/nn/__init__.pyú<module>r.      sD   ðñ 5õ
 #Ý /÷#÷ #÷ #÷ #÷ #÷ #÷ #÷ #÷ #ð #r,   