
    Vpf                    X   d Z ddlmZ ddlmZ ddlZddlZddlmZmZm	Z	 ddl
mZmZmZ ddlZej        ZeZej         G d de                      Ze	eee         ej        ef         Ze	eef         Zee         Z G d	 d
e          Z G d d          Z G d dej                  ZdS )a  
`jax._src.typing`: JAX type annotations
---------------------------------------

This submodule is a work in progress; when we finalize the contents here, it will be
exported at `jax.typing`. Until then, the contents here should be considered unstable
and may change without notice.

To see the proposal that led to the development of these tools, see
https://github.com/google/jax/pull/11859/.
    )annotations)SequenceN)AnyProtocolUnion)	ArrayLikeArrayStaticScalarc                  &    e Zd Zedd            ZdS )SupportsDTypereturnDTypec                    d S N selfs    O/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/typing.pydtypezSupportsDType.dtype1       C    Nr   r   )__name__
__module____qualname__propertyr   r   r   r   r   r   /   s&         8r   r   c                  >    e Zd Zedd            Zedd            ZdS )	DuckTypedArrayr   r   c                    d S r   r   r   s    r   r   zDuckTypedArray.dtypeH   r   r   Shapec                    d S r   r   r   s    r   shapezDuckTypedArray.shapeJ   r   r   Nr   )r   r    )r   r   r   r   r   r"   r   r   r   r   r   G   s<         8 8r   r   c                      e Zd Zd ZdS )DeprecatedArgc                    dS )N
Deprecatedr   r   s    r   __repr__zDeprecatedArg.__repr__Y   s    <r   N)r   r   r   r'   r   r   r   r$   r$   X   s#            r   r$   c                      e Zd ZdZdZdZdS )DLDeviceType      
   N)r   r   r   kDLCPUkDLCUDAkDLROCMr   r   r   r)   r)   ]   s        &''''r   r)   )__doc__
__future__r   collections.abcr   enumtypingr   r   r   jax._src.basearrayr   r	   r
   numpynpr   r   ExtendedDTyperuntime_checkabler   strtype	DTypeLikeintDimSizer    r   r$   IntEnumr)   r   r   r   <module>r@      s  
 
 # " " " " " $ $ $ $ $ $   ' ' ' ' ' ' ' ' ' '         
    
           H       s)(	 S/         X      "       
    4<     r   