sdpython / onnx-array-apiLinks
Array APIs to write ONNX Graphs
☆11Updated last month
Alternatives and similar repositories for onnx-array-api
Users that are interested in onnx-array-api are comparing it to the libraries listed below
Sorting:
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆32Updated 3 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆56Updated this week
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Pythonic framework for building ONNX graphs☆85Updated this week
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆28Updated 2 years ago
- ONNX Command-Line Toolbox☆35Updated 8 months ago
- Test pytorch code with minimal computational overhead☆25Updated 2 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆61Updated 2 months ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆49Updated 9 months ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- Hacks for PyTorch☆19Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆114Updated this week
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated last year
- Compatibility layer for common array libraries to support the Array API☆98Updated last week
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 7 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆48Updated 8 months ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Efficient in-memory representation for ONNX, in Python☆19Updated this week
- Convolutions and more as einsum for PyTorch☆16Updated last year
- Prototype routines for GPU quantization written using PyTorch.☆21Updated 4 months ago
- Scailable ONNX python tools☆97Updated 8 months ago
- PyTorch centric eager mode debugger☆47Updated 6 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆143Updated 4 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Write your code as tree-like expressions, then transform it☆21Updated last year
- Model compression for ONNX☆96Updated 7 months ago