sdpython / onnx-array-api

Related projects ⓘ

Alternatives and complementary repositories for onnx-array-api

• 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 2 months ago
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆27Updated 2 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆13Updated 3 weeks ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 2 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• Quantco / spox
  Pythonic framework for building ONNX graphs
  ☆74Updated this week
• onnx / neural-compressor
  Model compression for ONNX
  ☆75Updated last week
• zhenhuaw-me / onnxcli
  ONNX Command-Line Toolbox
  ☆35Updated last month
• aliutkus / groupmap
  GroupMap: beyond mean and variance matching for deep learning
  ☆10Updated 2 years ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆68Updated 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 4 months ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• oval-group / ali-g
  Implementation of the ALI-G algorithm (PyTorch, Tensorflow)
  ☆22Updated 3 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆23Updated this week
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆79Updated this week
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆26Updated last month
• tk-rusch / LEM
  Official code for Long Expressive Memory (ICLR 2022, Spotlight)
  ☆69Updated 2 years ago
• fpaupier / gRPC-multiprocessing
  A boilerplate to use multiprocessing for your gRPC server in your Python project
  ☆24Updated 3 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months ago
• josephrocca / onnxscript-editor
  ONNX Script editor & visualiser running completely in the browser thanks to Pyodide and Netron
  ☆19Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• tencent-ailab / TriNet
  TriNet: stabilizing self-supervised learning from complete or slow collapse on ASR.
  ☆26Updated last year