+
    NV-j¾   ã                   ó4   € ^ RI HtHt ^ RI5 ^ RIHtHtHt R# )é    )ÚinitÚlosses)Ú*)Úaverage_gradientsÚfsdp_apply_gradientsÚvalue_and_gradN)Úmlx.nnr   r   Úmlx.nn.layersÚmlx.nn.utilsr   r   r   © ó    Ú`/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx/nn/__init__.pyÚ<module>r      s   ð÷  Ü ÷ó r   