Anastasiia Filippova 503731727d QQ linear (#2931)

2026-01-05 11:20:54 -08:00

..

layers

QQ linear (#2931)

2026-01-05 11:20:54 -08:00

__init__.py

Ring docs (#1829)

2025-02-28 11:34:21 -08:00

init.py

Fix doc issues in mlx.nn.init.he_normal and mlx.nn.hard_tanh (#2968)

2026-01-05 07:23:41 -08:00

losses.py

fix cross entropy axis param (#2641)

2025-10-01 16:49:55 -07:00

utils.py

Remove stream from average grads so it uses default (#2532)

2025-08-25 15:56:29 -07:00