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☆66Updated last year
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆35Updated 3 weeks ago
- ONNX Command-Line Toolbox☆35Updated 11 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆59Updated 3 weeks ago
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆29Updated 2 years ago
- Model compression for ONNX☆97Updated 10 months ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆64Updated 5 months ago
- Pythonic framework for building ONNX graphs☆88Updated last week
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆123Updated 2 weeks ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years 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…☆181Updated last month
- ☆19Updated 2 years ago
- A simple Python tool to measure the performance of ONNX models.☆27Updated last year
- Customized matrix multiplication kernels☆56Updated 3 years ago
- This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…☆78Updated last week
- 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
- A Python library for Continual Inference Networks in PyTorch☆52Updated 6 months ago
- PyTorch ObjectDetection Modules and ONNX ops☆18Updated 2 years ago
- A toolbox to explore synchronous layerwise-parallel deep neural networks.☆17Updated 6 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
- Scailable ONNX python tools☆97Updated 11 months ago
- A faster implementation of OpenCV-CUDA that uses OpenCV objects, and more!☆53Updated last week
- PyCUDA based PyTorch Extension Made Easy☆26Updated last year
- Hacks for PyTorch☆19Updated 2 years ago
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated last year
- AI Edge Quantizer: flexible post training quantization for LiteRT models.☆67Updated last week
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Conversions between kornia and other computer vision libraries formats☆36Updated 2 years ago
- Script to check ONNX model compatibility against TensorRT versions using docker images☆12Updated last year