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
  ☆31Updated 4 months ago
• 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.
  ☆47Updated this week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆106Updated this week
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆82Updated 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
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆78Updated this week
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆122Updated 3 weeks ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆41Updated 2 months ago
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆19Updated last month
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 4 months ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated 3 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆240Updated this week
• ibaiGorordo / onnx-perf-test
  A simple Python tool to measure the performance of ONNX models.
  ☆26Updated 4 months ago
• aliutkus / groupmap
  GroupMap: beyond mean and variance matching for deep learning
  ☆10Updated 2 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆18Updated last year
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆23Updated last month
• lucidrains / light-recurrent-unit-pytorch
  Implementation of a Light Recurrent Unit in Pytorch
  ☆48Updated 3 months ago
• PINTO0309 / sne4onnx
  A very simple tool for situations where optimization with onnx-simplifier would exceed the Protocol Buffers upper file size limit of 2GB,…
  ☆15Updated 8 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated 3 months ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆37Updated 4 months ago
• onnx / neural-compressor
  Model compression for ONNX
  ☆81Updated 2 months ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 2 months ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 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.
  ☆43Updated 6 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆44Updated last month