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 / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆53Updated this week
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆27Updated 2 years ago
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆93Updated last week
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated last year
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 7 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆109Updated last week
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆81Updated last week
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated 6 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆54Updated 3 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆53Updated this week
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆138Updated 2 months ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 5 months ago
• guillaumeBellec / multitask
  ☆22Updated 5 months ago
• pytorch / maskedtensor
  MaskedTensors for PyTorch
  ☆38Updated 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
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• conda-forge / nvcc-feedstock
  A conda-smithy repository for nvcc.
  ☆12Updated 2 months ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated 10 months ago
• microsoft / pytorch_od
  PyTorch ObjectDetection Modules and ONNX ops
  ☆18Updated last year
• akshayka / hessian_trace_estimation
  Hessian trace estimation using PyTorch and Hutch++
  ☆19Updated 4 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 4 months ago
• Quansight-Labs / numpy_pytorch_interop
  ☆20Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago