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:
- Productionize machine learning predictions, with ONNX or without☆66Updated last year
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆34Updated 3 weeks ago
- ONNX Command-Line Toolbox☆35Updated 10 months ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆63Updated 4 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆59Updated this week
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆29Updated 2 years ago
- Model compression for ONNX☆97Updated 9 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated 3 weeks 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…☆180Updated last month
- Scailable ONNX python tools☆97Updated 10 months ago
- This library empowers users to seamlessly port pretrained models and checkpoints on the HuggingFace (HF) hub (developed using HF transfor…☆75Updated this week
- Pythonic framework for building ONNX graphs☆88Updated last week
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- Hacks for PyTorch☆19Updated 2 years ago
- ☆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 year
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- PyTorch ObjectDetection Modules and ONNX ops☆18Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
- A Python library for Continual Inference Networks in PyTorch☆52Updated 5 months ago
- A toolbox to explore synchronous layerwise-parallel deep neural networks.☆17Updated 6 years ago
- ☆36Updated 3 years ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆66Updated 2 weeks ago
- A conda-smithy repository for nvcc.☆13Updated 7 months ago
- PyTorch centric eager mode debugger☆48Updated 8 months ago
- Code for the simulations in the neural Kalman filtering paper☆20Updated 4 years ago
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year