o
    b¦µi4  ã                   @   sJ   d dl mZ eƒ rd dlmZ G dd„ dejƒZneZG dd„ deƒZdS )é    )Úis_nnx_enabled)Únnxc                       s   e Zd Z‡ fdd„Z‡  ZS )Ú	BaseLayerc                    s   t ƒ jdddi|¤Ž d S )NÚpytreeF© )ÚsuperÚ__init_subclass__)ÚclsÚkwargs©Ú	__class__r   úO/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/backend/jax/layer.pyr      s   zBaseLayer.__init_subclass__)Ú__name__Ú
__module__Ú__qualname__r   Ú__classcell__r   r   r   r   r      s    r   c                   @   s   e Zd ZdS )ÚJaxLayerN)r   r   r   r   r   r   r   r      s    r   N)Úkeras.src.backend.configr   Úflaxr   ÚModuler   Úobjectr   r   r   r   r   Ú<module>   s    