Al0den / metalgpuLinks
A python interface for Apple GPU's Metal API
☆27Updated 6 months ago
Alternatives and similar repositories for metalgpu
Users that are interested in metalgpu are comparing it to the libraries listed below
Sorting:
- A python library to run metal compute kernels on macOS☆88Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated last week
- Minimal yet performant LLM examples in pure JAX☆240Updated 3 weeks ago
- JAX-Toolbox☆382Updated this week
- Minimal, lightweight JAX implementations of popular models.☆187Updated last week
- Nabla is a novel Distributed-Tensor and Scientific-Computing Framework; built from scratch on top of Mojo and MAX☆318Updated this week
- ☆91Updated last year
- Orbax provides common checkpointing and persistence utilities for JAX users☆480Updated this week
- 🧱 Modula software package☆322Updated 5 months ago
- OpTree: Optimized PyTree Utilities☆205Updated last week
- Efficient framework-agnostic data loading☆459Updated 4 months ago
- FlashAttention (Metal Port)☆579Updated last year
- Run PyTorch in JAX. 🤝☆312Updated 3 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆329Updated this week
- JAX bindings for Flash Attention v2☆103Updated this week
- a Jax quantization library☆90Updated this week
- JMP is a Mixed Precision library for JAX.☆211Updated last year
- ☆70Updated last year
- ☆265Updated last week
- Pytorch-like dataloaders for JAX.☆99Updated last month
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆437Updated last month
- Getting crystal-like representations with harmonic loss☆195Updated 10 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆158Updated 2 months ago
- NVIDIA Math Libraries for the Python Ecosystem☆544Updated 3 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆306Updated last year
- Graph Neural Network library made for Apple Silicon☆207Updated 5 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆536Updated last month
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆155Updated 2 years ago
- Running Jax in PyTorch Lightning☆119Updated last year