baldand / py-metal-compute
A python library to run metal compute kernels on macOS
☆68Updated last year
Related projects ⓘ
Alternatives and complementary repositories for py-metal-compute
- Emulating double-precision arithmetic on Apple GPUs☆47Updated last year
- Python + Apple Metal Framework = ?☆35Updated 5 years ago
- Python interface to VkFFT☆51Updated last month
- Software library for FDTD of viscoelastic equation using a staggered grid arrangement with support for GPU and CPU backends☆54Updated 4 months ago
- 1D, 2D, and 3D variations of Fast Fourier Transforms☆28Updated 2 years ago
- Compatibility layer for common array libraries to support the Array API☆79Updated this week
- Study and Implementations of Numerical Algorithms on Apple M1 and A* Devices☆124Updated last year
- Print all known information about the GPU on Apple-designed chips☆64Updated 2 months ago
- An experimental repo for accessing Metal API from Python (OSX Only)☆22Updated 4 years ago
- Metal Shading Language on Apple M1's GPU for scientific C++.☆82Updated last year
- Running linear algebra as fast as possible on Apple silicon☆18Updated last year
- Renderer for molecular nanotechnology☆65Updated 2 months ago
- Nonuniform fast Fourier transforms of types 1 and 2, in 1D, 2D, and 3D, on the GPU☆83Updated 11 months ago
- C API for MLX☆77Updated this week
- NVIDIA Math Libraries for the Python Ecosystem☆205Updated this week
- Swift Binding for MuJoCo: https://mujoco.org/☆18Updated last year
- ☆47Updated 3 years ago
- Run a CoreML MLModel on the Asahi Neural Engine☆45Updated last year
- FlashAttention (Metal Port)☆386Updated last month
- Library to manipulate Apple Metal Shading Language IR☆48Updated last year
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆215Updated this week
- RFC document, tooling and other content related to the array API standard☆215Updated 3 weeks ago
- Exploring using stdpar and Cython☆32Updated 4 years ago
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- A nanobind example project☆91Updated last week
- WebGPU for Python☆466Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆27Updated this week
- mlx image models for Apple Silicon machines☆68Updated 6 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated 2 months ago