Angelos Katharopoulos 53e6a9367c Use reshape and transpose for non-overlapping pooling windows (#867)

2024-03-21 10:21:03 -07:00

..

mlx

Use reshape and transpose for non-overlapping pooling windows (#867)

2024-03-21 10:21:03 -07:00

src

Fast RMS Norm (#862)

2024-03-21 07:20:54 -07:00

tests

Add minimum for cosine decay function (#859)

2024-03-21 07:33:29 -07:00