KAN (Kolmogorov–Arnold Networks) in the MLX framework - View it on GitHub