Alternatives and similar repositories for onnx-array-api

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

• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆57Updated this week
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• sdpython / onnx-extended
  New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA
  ☆32Updated 2 months ago
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆84Updated this week
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆60Updated last month
• 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.
  ☆110Updated 3 weeks ago
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆28Updated 2 years ago
• Quansight-Labs / numpy_pytorch_interop
  ☆20Updated last year
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆26Updated this week
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆21Updated 3 months ago
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆97Updated 2 weeks ago
• buttercutter / Mamba_SSM
  A simple implementation of [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/abs/2312.00752)
  ☆21Updated last year
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated 7 months ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• 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.
  ☆44Updated 10 months ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• onnx / ir-py
  Efficient in-memory representation for ONNX, in Python
  ☆13Updated this week
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 8 months ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 8 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆23Updated 2 weeks ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated last year
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated last year
• lucidrains / light-recurrent-unit-pytorch
  Implementation of a Light Recurrent Unit in Pytorch
  ☆47Updated 8 months ago
• google-ai-edge / ai-edge-quantizer
  AI Edge Quantizer: flexible post training quantization for LiteRT models.
  ☆41Updated last week
• quic / efficient-transformers
  This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…
  ☆68Updated this week
• onnx / neural-compressor
  Model compression for ONNX
  ☆96Updated 6 months ago