search

ml-explore / mlx

MLX: An array framework for Apple silicon
https://ml-explore.github.io/mlx/
MIT License
14.83k stars 845 forks source link

[Feature] KANs #1068

Closed Stealeristaken closed 2 weeks ago

Stealeristaken commented 2 weeks ago

Should we integrate the recent Kolmogorov Arnold Neural Network (KAN) structure into MLX? Or is it better to wait?

KANs -> https://arxiv.org/pdf/2404.19756

awni commented 2 weeks ago

That would be cool to see! There's no need to wait to try it, though it should be done in a separate repo. Let us know if any features are missing!