bitorch_engine.utils.mlx_extension

Functions

get_mlx_extension

Creates and returns a CppExtension for compiling a C++ extension with MLX library support.