
    h                        S SK JrJrJrJrJrJrJrJrJ	r	J
r
Jr  S SKJrJrJr  S SKJrJrJr  S SKJrJrJrJrJr  S SKJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;J<r<J=r=J>r>J?r?J@r@JArAJBrBJCrCJDrDJErEJFrFJGrGJHrHJIrIJJrJJKrKJLrLJMrMJNrNJOrOJPrPJQrQJRrRJSrSJTrTJUrUJVrVJWrWJXrXJYrYJZrZJ[r[J\r\J]r]J^r^J_r_J`r`JaraJbrbJcrcJdrdJereJfrfJgrgJhrhJiriJjrjJkrk  S SKlJmrmJnrnJoroJprp  S SKqJrrrJsrsJtrtJuruJvrvJwrw  S SKxJyryJzrzJ{r{J|r|  S S	K}J~r~JrJrJrJrJrJr  S S
KJrJrJrJrJrJrJrJr  S SKJrJrJrJr  S SKJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJr  / SQrg)   )CupyOpsMPSOpsNumpyOpsOpsget_current_opsget_opsset_current_opsset_gpu_allocatoruse_opsuse_pytorch_for_gpu_memoryuse_tensorflow_for_gpu_memory)enable_mxnetenable_tensorflowhas_cupy)ConfigConfigValidationErrorregistry)configure_normal_initglorot_uniform_initnormal_inituniform_init	zero_init)QLSTMCauchySimilarityClippedLinearDishDropoutEmbedGeluHardSigmoid	HardSwishHardSwishMobilenetHardTanh	HashEmbed	LayerNormLinearLogisticMaxoutMishMultiSoftmaxMXNetWrapperParametricAttentionParametricAttention_v2PyTorchLSTMPyTorchRNNWrapperPyTorchWrapperPyTorchWrapper_v2PyTorchWrapper_v3ReluReluKSigmoidSoftmax
Softmax_v2SparseLinearSparseLinear_v2SwishTensorFlowWrapperTorchScriptWrapper_v1addarray_getitembidirectionalchaincloneconcatenateexpand_windowkeras_subclass
list2arraylist2paddedlist2raggedmap_listnooppadded2list
premap_idspytorch_to_torchscript_wrapperragged2listreduce_firstreduce_last
reduce_maxreduce_mean
reduce_sum	remap_idsremap_ids_v2residual	resizablesiamesesigmoid_activationsoftmax_activationstrings2arraystuplifyuniqued
with_arraywith_array2dwith_cpu
with_debugwith_flattenwith_flatten_v2with_getitem	with_listwith_nvtx_rangewith_paddedwith_raggedwith_reshapewith_signpost_interval)CategoricalCrossentropyCosineDistance
L2DistanceSequenceCategoricalCrossentropy)Modelchange_attr_valuesdeserialize_attrserialize_attrset_dropout_ratewrap_model_recursive)SGDAdam	OptimizerRAdam)compoundingconstantconstant_thencyclic_triangulardecayingslanted_triangularwarmup_linear)	MXNetShimPyTorchGradScalerPyTorchShimShimTensorFlowShimTorchScriptShimkeras_model_fnsmaybe_handshake_model)
ArgsKwargsPaddedRaggedUnserializable)DataValidationErrordata_validationfix_random_seedget_array_moduleget_torch_default_device	get_widthis_cupy_arraymxnet2xp
prefer_gpurequire_cpurequire_gpuset_active_gputensorflow2xpto_categoricalto_numpytorch2xpxp2mxnetxp2tensorflowxp2torch)r   r   r   r   r   r   r   r   rj   rl   rk   rm   rn   rq   rp   rr   ro   rs   r   r   r   r   r   r   r   r   ru   rw   rt   rv   r{   r~   ry   rz   r|   r}   rx   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   r   r
   r   r   r   r   rC   r$   r%   r&   r(   r)   r*   r3   rY   r6   r   r   r,   r'   rV   rX   r5   r8   r   r4   r#   r    r   r!   r"   r:   r   r0   r/   r.   r;   rD   r+   r1   r7   r2   r9   r<   r-   r=   r?   r@   rA   rB   rI   rU   r\   rW   rG   rM   rH   r]   r^   rf   rd   rg   ra   rh   rc   rZ   rE   rG   rM   rF   rJ   rS   rT   rK   r>   r_   r`   re   ri   r[   rb   rL   rN   rO   rP   rQ   rR   N)backendsr   r   r   r   r   r   r	   r
   r   r   r   compatr   r   r   configr   r   r   initializersr   r   r   r   r   layersr   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   lossrj   rk   rl   rm   modelrn   ro   rp   rq   rr   rs   
optimizersrt   ru   rv   rw   	schedulesrx   ry   rz   r{   r|   r}   r~   shimsr   r   r   r   r   r   r   r   typesr   r   r   r   utilr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   __all__     C/home/james-whalen/.local/lib/python3.13/site-packages/thinc/api.py<module>r      s      > = ; ; R R R R R R R R R R R R R R R R R R R R Rf   4 3  	 	 	 > =     .?r   