sdpython / onnx-array-apiLinks
Array APIs to write ONNX Graphs
☆11Updated 2 months ago
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:
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆33Updated this week
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆28Updated 2 years ago
- ONNX Command-Line Toolbox☆35Updated 9 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆57Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆115Updated 3 weeks ago
- A Python library for Continual Inference Networks in PyTorch☆52Updated 4 months ago
- Test pytorch code with minimal computational overhead☆25Updated 2 years ago
- Model compression for ONNX☆96Updated 8 months ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆65Updated 2 weeks ago
- A very simple tool for situations where optimization with onnx-simplifier would exceed the Protocol Buffers upper file size limit of 2GB,…☆17Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Pythonic framework for building ONNX graphs☆86Updated this week
- ☆18Updated 2 years ago
- Scailable ONNX python tools☆97Updated 8 months ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 8 months ago
- Convolutions and more as einsum for PyTorch☆17Updated last year
- Customized matrix multiplication kernels☆56Updated 3 years ago
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆70Updated 3 years ago
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Hacks for PyTorch☆19Updated 2 years ago
- A faster implementation of OpenCV-CUDA that uses OpenCV objects, and more!☆51Updated last week
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated last week
- Conversions between kornia and other computer vision libraries formats☆36Updated last year
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆20Updated 3 years ago
- C++ implementations for various tokenizers (sentencepiece, tiktoken etc).☆32Updated this week
- ☆8Updated last year