Alternatives and similar repositories for onnx-array-api

Users that are interested in onnx-array-api are comparing it to the libraries listed below

• sdpython / onnx-extended
  New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA
  ☆32Updated last month
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆59Updated 3 weeks ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆55Updated this week
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 6 months ago
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated 7 months ago
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆27Updated 2 years ago
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆81Updated last week
• ibaiGorordo / onnx-perf-test
  A simple Python tool to measure the performance of ONNX models.
  ☆26Updated 7 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆111Updated this week
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• aliutkus / groupmap
  GroupMap: beyond mean and variance matching for deep learning
  ☆10Updated 2 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated 2 years ago
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆21Updated 2 months ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• npuichigo / blazing-fast-io-tutorial
  Blazing fast data loading with HuggingFace Dataset and Ray Data
  ☆16Updated last year
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 7 months ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆40Updated 8 months ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• scailable / sclblonnx
  Scailable ONNX python tools
  ☆97Updated 6 months ago
• guillaumeBellec / multitask
  ☆22Updated 6 months ago
• conda-forge / nvcc-feedstock
  A conda-smithy repository for nvcc.
  ☆12Updated 3 months ago
• onnx / neural-compressor
  Model compression for ONNX
  ☆92Updated 5 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆54Updated 3 years ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆45Updated 9 months ago
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated last year
• 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…
  ☆180Updated 5 months ago