Alternatives and similar repositories for nvmath-python:

Users that are interested in nvmath-python are comparing it to the libraries listed below

• nv-legate / legate
  The Foundation for All Legate Libraries
  ☆202Updated last month
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆240Updated this week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆279Updated this week
• NVIDIA / accelerated-computing-hub
  NVIDIA curated collection of educational resources related to general purpose GPU programming.
  ☆175Updated last week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆294Updated this week
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆332Updated last week
• NVIDIA / numba-cuda
  The CUDA target for Numba
  ☆43Updated this week
• IntelPython / dpnp
  Data Parallel Extension for NumPy
  ☆101Updated this week
• salykova / sgemm.cu
  SGEMM that beats cuBLAS
  ☆68Updated last week
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆550Updated 2 months ago
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆457Updated last week
• pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆178Updated last month
• NVIDIA / numbast
  Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.
  ☆32Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆525Updated this week
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆328Updated this week
• KernelTuner / kernel_tuner
  Kernel Tuner
  ☆304Updated 2 weeks ago
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆383Updated 5 months ago
• NVIDIA / cuCollections
  ☆506Updated this week
• nv-legate / cupynumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆821Updated 3 weeks ago
• intel / intel-extension-for-openxla
  ☆41Updated this week
• NVIDIA / nvidia-hpcg
  NVIDIA HPCG is based on the HPCG benchmark and optimized for performance on NVIDIA accelerated HPC systems.
  ☆46Updated 3 months ago
• IntelPython / numba-dpex
  Data Parallel Extension for Numba
  ☆79Updated 2 months ago
• rapidsai / kvikio
  KvikIO - High Performance File IO
  ☆176Updated this week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆105Updated this week
• eth-cscs / COSMA
  Distributed Communication-Optimal Matrix-Matrix Multiplication Algorithm
  ☆198Updated last month
• spcl / npbench
  NPBench - A Benchmarking Suite for High-Performance NumPy
  ☆76Updated 2 months ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆84Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆371Updated this week
• pytorch / torchft
  PyTorch per step fault tolerance (actively under development)
  ☆226Updated this week
• NVIDIA / TorchFort
  An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
  ☆159Updated this week