onnx / ir-pyLinks
Efficient in-memory representation for ONNX, in Python
☆42Updated this week
Alternatives and similar repositories for ir-py
Users that are interested in ir-py are comparing it to the libraries listed below
Sorting:
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆49Updated 5 months ago
- Visualize ONNX models with model-explorer☆67Updated last month
- ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.☆420Updated last week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆329Updated this 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…☆182Updated last month
- Model compression for ONNX☆98Updated last year
- QONNX: Arbitrary-Precision Quantized Neural Networks in ONNX☆175Updated last week
- ☆137Updated last week
- MLIR-based partitioning system☆164Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- High-Performance FP32 GEMM on CUDA devices☆117Updated last year
- TORCH_TRACE parser for PT2☆76Updated this week
- Machine Learning Agility (MLAgility) benchmark and benchmarking tools☆40Updated 6 months ago
- A Data-Centric Compiler for Machine Learning☆85Updated last month
- Common utilities for ONNX converters☆294Updated last month
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- python package of rocm-smi-lib☆24Updated last month
- Memory Optimizations for Deep Learning (ICML 2023)☆115Updated last year
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆104Updated last month
- This repository is a read-only mirror of https://gitlab.arm.com/kleidi/kleidiai☆113Updated this week
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- A lightweight, Pythonic, frontend for MLIR☆81Updated 2 years ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- OpenAI Triton backend for Intel® GPUs☆226Updated this week
- ☆344Updated this week
- Digest AI is a powerful model analysis tool that extracts insights from your models.☆40Updated 8 months ago
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆105Updated this week
- Notes and artifacts from the ONNX steering committee☆28Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆164Updated 3 weeks ago
- LLM training in simple, raw C/CUDA☆112Updated last year