Angelos Katharopoulos 1944cf67a2 Add vmap for BroadcastAxes (#3344)

2026-03-31 17:08:56 -07:00

..

mlx

Fix repr of conv layers (#3275)

2026-03-18 22:47:38 -07:00

src

Fix np bfloat16 misinterpreted as complex (#3146)

2026-03-31 08:04:55 +09:00

tests

Add vmap for BroadcastAxes (#3344)

2026-03-31 17:08:56 -07:00