Maintainer of Nx
We should hopefully be releasing the library in the next few days. We're just missing a really small set of Nx.Backend calbacks and some housekeeping stuff
We should hopefully be releasing the library in the next few days. We're just missing a really small set of Nx.Backend calbacks and some housekeeping stuff
A bit slower than Nx.BinaryBackend (runs the loop in 5s), but this is a simple model which really doesn't take advantage of a native platform.
Next target is to get full Nx API coverage so we can run Bumblebee on it.
A bit slower than Nx.BinaryBackend (runs the loop in 5s), but this is a simple model which really doesn't take advantage of a native platform.
Next target is to get full Nx API coverage so we can run Bumblebee on it.
Jokes aside, big shout out to @samrat.me and @uwucocoa.moe who are taking MLX coverage by stride.
We're almost at a point where we can run an Axon model!
Bonus: this will be plug and play with Sharding
Jokes aside, big shout out to @samrat.me and @uwucocoa.moe who are taking MLX coverage by stride.
We're almost at a point where we can run an Axon model!
Bonus: this will be plug and play with Sharding