There is also a MLX implementation: https://github.com/awni/voxmlx