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 3 months ago
• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆56Updated this week
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆85Updated this week
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆28Updated 2 years ago
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated 8 months ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated 2 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆61Updated 2 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 9 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆56Updated 3 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated 2 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆114Updated this week
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated last year
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆98Updated last week
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 7 months ago
• lucidrains / light-recurrent-unit-pytorch
  Implementation of a Light Recurrent Unit in Pytorch
  ☆48Updated 8 months ago
• akshayka / hessian_trace_estimation
  Hessian trace estimation using PyTorch and Hutch++
  ☆19Updated 4 years ago
• onnx / ir-py
  Efficient in-memory representation for ONNX, in Python
  ☆19Updated this week
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated last year
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆21Updated 4 months ago
• scailable / sclblonnx
  Scailable ONNX python tools
  ☆97Updated 8 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆47Updated 6 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆143Updated 4 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• onnx / neural-compressor
  Model compression for ONNX
  ☆96Updated 7 months ago