mlx/python
Awni Hannun 58d0e199e1
add bfloat conv for windograd (#1306)
* add bfloat conv for windograd

* accumulate in fp32

* accumulate in fp32

* accumulate in bf16
2024-08-05 15:51:13 -07:00
..
mlx faster group norm (#1304) 2024-08-01 12:49:23 -07:00
src fix creating array from bf16 tensors in jax / torch (#1305) 2024-08-01 16:20:51 -07:00
tests add bfloat conv for windograd (#1306) 2024-08-05 15:51:13 -07:00