You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
We can copy most of that workflow to get a similarly-compiled defn function in EMLX.compile. EMLX can get the (vector) -> vector function pointer as a NIF resource, cache that in the compiler key and use it for the Elixir function as well.
The text was updated successfully, but these errors were encountered:
MLX uses an abstraction for compiling Python functions: https://github.com/ml-explore/mlx/blob/974bb54ab2f8e8450f856ac215df160374080b33/python/src/transforms.cpp#L370C8-L370C21
We can copy most of that workflow to get a similarly-compiled defn function in EMLX.compile. EMLX can get the (vector) -> vector function pointer as a NIF resource, cache that in the compiler key and use it for the Elixir function as well.
The text was updated successfully, but these errors were encountered: