k-search-4ba9bdda·1 events·first seen Aliases: K-Search
Researchers at UC Berkeley extended K-Search, an evolutionary AI-driven kernel optimization framework, with a CUDA-to-MLX translation layer that adapts existing CUDA kernels to Apple Silicon rather than rebuilding from scratch. The system uses an LLM (Gemini 3.5 Pro Preview) to maintain a structured 'world model' decision tree and iteratively generate, compile, and benchmark candidate MLX kernels. Results show 0.97x performance relative to the native MLX Attention kernel and up to 20x prefill speedup over the community mlx-lm implementation on the Mamba SSM kernel. The approach is framed as generally applicable to any ecosystem where CUDA expertise is transferable, not just MLX.