
    ij4                     p    d dl mZ  e       rd dlmZ  G d dej                        ZneZ G d de      Zy)    )is_nnx_enabled)nnxc                        e Zd Z fdZ xZS )	BaseLayerc                 (    t        |   dddi| y )NpytreeF )super__init_subclass__)clskwargs	__class__s     p/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/backend/jax/layer.pyr   zBaseLayer.__init_subclass__   s    G%=U=f=    )__name__
__module____qualname__r   __classcell__)r   s   @r   r   r      s    	> 	>r   r   c                       e Zd Zy)JaxLayerN)r   r   r   r	   r   r   r   r      s    r   r   N)keras.src.backend.configr   flaxr   Moduler   objectr   r	   r   r   <module>r      s4    3>CJJ > I	y 	r   