Angelos Katharopoulos 1a87dc5ea8 Fix compile fusion for multi-output edge cases (#950) ...

* Fix compile fusion for multi-output edge cases

* Add a test for multi-output compile

2024-04-02 08:42:31 -07:00

..

mlx

Support Chaining for some of functionalities of nn.Module (#885) (#897)

2024-03-27 19:58:29 -07:00

src

Indexing bug fix (#947)

2024-04-01 12:18:50 -07:00

tests

Fix compile fusion for multi-output edge cases (#950)

2024-04-02 08:42:31 -07:00