Alternatives and similar repositories for metalgpu

Users that are interested in metalgpu are comparing it to the libraries listed below

• baldand / py-metal-compute
  A python library to run metal compute kernels on macOS
  ☆88Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆130Updated last week
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆240Updated 3 weeks ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆382Updated this week
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆187Updated last week
• nabla-ml / nabla
  Nabla is a novel Distributed-Tensor and Scientific-Computing Framework; built from scratch on top of Mojo and MAX
  ☆318Updated this week
• vdesai2014 / inference-optimization-blog-post
  ☆91Updated last year
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆480Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆205Updated last week
• ml-explore / mlx-data
  Efficient framework-agnostic data loading
  ☆459Updated 4 months ago
• philipturner / metal-flash-attention
  FlashAttention (Metal Port)
  ☆579Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆312Updated 3 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆329Updated this week
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆103Updated this week
• google / qwix
  a Jax quantization library
  ☆90Updated this week
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated last year
• apple / ml-ademamix
  ☆70Updated last year
• jax-ml / jax-ai-stack
  ☆265Updated last week
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆99Updated last month
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆513Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆437Updated last month
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆195Updated 10 months ago
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆158Updated 2 months ago
• NVIDIA / nvmath-python
  NVIDIA Math Libraries for the Python Ecosystem
  ☆544Updated 3 weeks ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆306Updated last year
• mlx-graphs / mlx-graphs
  Graph Neural Network library made for Apple Silicon
  ☆207Updated 5 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆536Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆155Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆119Updated last year