sdpython / onnx-array-apiLinks
Array APIs to write ONNX Graphs
☆11Updated 2 weeks 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:
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆35Updated 2 weeks ago
 - Productionize machine learning predictions, with ONNX or without☆66Updated last year
 - ONNX-backed array library that is compliant with the Array API standard.☆60Updated this week
 - This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…☆82Updated this week
 - 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
 - Model compression for ONNX☆98Updated 11 months ago
 - Pythonic framework for building ONNX graphs☆88Updated this week
 - AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆28Updated 3 years ago
 - 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.☆123Updated last month
 - No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆67Updated 6 months ago
 - A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
 - Scailable ONNX python tools☆97Updated last year
 - 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 2 months ago
 - A Python library for Continual Inference Networks in PyTorch☆53Updated 7 months ago
 - Customized matrix multiplication kernels☆57Updated 3 years ago
 - Repository containing the Kornia related tutorials☆47Updated 2 months ago
 - ☆19Updated 3 years ago
 - Hacks for PyTorch☆19Updated 2 years 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
 - TorchFix - a linter for PyTorch-using code with autofix support☆148Updated 2 months ago
 - A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆65Updated 2 weeks ago
 - A toolbox to explore synchronous layerwise-parallel deep neural networks.☆17Updated 6 years ago
 - Triton Server Component for lightning.ai☆14Updated 2 years ago
 - Framework for writing deep learning training loops. Lightweight, and retaining full freedom to design as you see fits. It handles checkpo…☆117Updated last year
 - Conversions between kornia and other computer vision libraries formats☆36Updated 2 years ago
 - PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆17Updated 3 years ago
 - Nonlinear solver backend for PyTorch☆68Updated 6 years ago
 - Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago