ThunderMittens is an Apple Metal Shading Language (MSL) port for the ThunderKittens framework.

The repository supports two primary use cases:

For writing Metal Shading Language (MSL) kernels:

• Clone the ThunderKittens repository
• Open the project in Xcode
• Xcode will handle all build processes

For using ThunderKittens kernels within MLX in Python:

• Python 3.8+
• CMake
• Xcode Command Line Tools

1. Navigate to ThunderMittens/mlx directory

2. Install MLX with parallel build:

   CMAKE_BUILD_PARALLEL_LEVEL=8 pip install -e ".[dev]"

3. Navigate to ThunderMittens/kernels directory

4. Install requirements:

   pip install -r requirements.txt

5. Build kernels and bindings:

   python setup.py build_ext -j8 --inplace

Please see our blog post to learn more about this work. Please checkout our paper to learn more about the ThunderKittens project.