nv-legate / cupynumeric

Related projects ⓘ

Alternatives and complementary repositories for cupynumeric

• nv-legate / legate
  The Foundation for All Legate Libraries
  ☆193Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆912Updated last month
• dionhaefner / pyhpc-benchmarks
  A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python
  ☆312Updated last month
• NVIDIA / cuda-python
  CUDA Python Low-level Bindings
  ☆965Updated this week
• dgasmith / opt_einsum
  ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
  ☆863Updated last month
• NVIDIA / multi-gpu-programming-models
  Examples demonstrating available options to program multiple GPUs in a single node or a cluster
  ☆558Updated 3 weeks ago
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆492Updated this week
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆378Updated 3 months ago
• NVIDIA / nvmath-python
  NVIDIA Math Libraries for the Python Ecosystem
  ☆205Updated this week
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆445Updated 2 weeks ago
• nv-legate / legate.pandas
  An Aspiring Drop-In Replacement for Pandas at Scale
  ☆74Updated 3 years ago
• KernelTuner / kernel_tuner
  Kernel Tuner
  ☆287Updated last week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,396Updated this week
• NVIDIA / cuCollections
  ☆486Updated this week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,220Updated this week
• NVIDIA / AMGX
  Distributed multigrid linear solver library on GPU
  ☆495Updated 3 months ago
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,054Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,011Updated 7 months ago
• rapidsai / dask-cuda
  Utilities for Dask and CUDA interactions
  ☆292Updated this week
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆589Updated this week
• EnzymeAD / Enzyme
  High-performance automatic differentiation of LLVM and MLIR.
  ☆1,287Updated this week
• mpi4py / mpi4py
  Python bindings for MPI
  ☆814Updated this week
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,256Updated 7 months ago
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆933Updated 2 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,699Updated last week
• lebedov / scikit-cuda
  Python interface to GPU-powered libraries
  ☆987Updated last year
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆495Updated 3 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆249Updated last year
• NVIDIA / hpc-container-maker
  HPC Container Maker
  ☆457Updated 2 weeks ago
• inducer / pycuda
  CUDA integration for Python, plus shiny features
  ☆1,853Updated 2 weeks ago