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☆35Updated 3 weeks ago
- Productionize machine learning predictions, with ONNX or without☆66Updated last year
- ONNX Command-Line Toolbox☆35Updated 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☆90Updated 3 weeks ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆69Updated 7 months ago
- Model compression for ONNX☆99Updated last year
- ☆19Updated 3 years ago
- Scailable ONNX python tools☆97Updated last year
- A simple Python tool to measure the performance of ONNX models.☆27Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated 2 months ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆182Updated 3 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆61Updated 3 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 month
- Customized matrix multiplication kernels☆57Updated 3 years ago
- A Python library for Continual Inference Networks in PyTorch☆53Updated 8 months ago
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆29Updated 3 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
- AI Edge Quantizer: flexible post training quantization for LiteRT models.☆78Updated last week
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…☆83Updated this week
- Loop Nest - Linear algebra compiler and code generator.☆21Updated 3 years ago
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆71Updated 3 years ago
- PyTorch ObjectDetection Modules and ONNX ops☆18Updated 2 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆151Updated 3 months ago
- Online Normalization for Training Neural Networks (Companion Repository)☆86Updated 4 years ago
- Framework for writing deep learning training loops. Lightweight, and retaining full freedom to design as you see fits. It handles checkpo…☆116Updated last year
- Hacks for PyTorch☆19Updated 2 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆312Updated this week
- ONNX and TensorRT implementation of Whisper☆65Updated 2 years ago