sdpython / onnx-array-api

Related projects ⓘ

Alternatives and complementary repositories for onnx-array-api

• sdpython / onnx-extended
  New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA
  ☆28Updated last month
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆66Updated 10 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated last month
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆72Updated this week
• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆40Updated this week
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated last month
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆27Updated 2 years ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆48Updated last month
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆60Updated this week
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆35Updated 3 months ago
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆76Updated this week
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• onnx / neural-compressor
  Model compression for ONNX
  ☆74Updated last month
• Quansight-Labs / numpy_pytorch_interop
  ☆19Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated this week
• aliutkus / groupmap
  GroupMap: beyond mean and variance matching for deep learning
  ☆10Updated 2 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆20Updated last week
• morousg / cvGPUSpeedup
  A faster implementation of OpenCV-CUDA that uses OpenCV objects, and more!
  ☆42Updated last week
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆66Updated last year
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• josephrocca / onnxscript-editor
  ONNX Script editor & visualiser running completely in the browser thanks to Pyodide and Netron
  ☆19Updated last year
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆13Updated last week
• tk-rusch / LEM
  Official code for Long Expressive Memory (ICLR 2022, Spotlight)
  ☆69Updated 2 years ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆100Updated this week
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated this week