Quansight-Labs / numpy_pytorch_interop

Related projects ⓘ

Alternatives and complementary repositories for numpy_pytorch_interop

• iree-org / iree-jax
  ☆48Updated 3 months ago
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆25Updated last week
• Jokeren / triton-samples
  ☆23Updated 2 months ago
• 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
• NVIDIA / numba-cuda
  ☆35Updated this week
• FrancescoSaverioZuppichini / pytorch-2.0-benchmark
  Benchmarking PyTorch 2.0 different models
  ☆21Updated last year
• YichengDWu / yoho
  A compiler written in Mojo 🔥 and generates RISC-V assembly
  ☆14Updated 4 months ago
• jax-ml / jax-ai-stack
  ☆65Updated this week
• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated 2 weeks ago
• nv-legate / legate.pandas
  An Aspiring Drop-In Replacement for Pandas at Scale
  ☆73Updated 3 years ago
• data-apis / array-api-comparison
  Data and tooling to compare the API surfaces of various array libraries.
  ☆54Updated 5 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆208Updated this week
• rsokl / phantom-tensors
  Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing
  ☆17Updated last week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated last month
• nod-ai / PI
  A lightweight MLIR Python frontend with support for PyTorch
  ☆20Updated 2 months ago
• Quantco / ndonnx
  ONNX-backed array library that is compliant with the Array API standard.
  ☆39Updated this week
• suo / lintrunner
  ☆24Updated 5 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• nebari-dev / nebari-slurm
  An opinionated open source deployment of jupyterhub based on an Slurm job scheduler.
  ☆28Updated last month
• DPBayes / jax-chacha-prng
  A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.
  ☆12Updated 8 months ago
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆30Updated last year
• probml / JSL
  Jax SSM Library
  ☆50Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆82Updated this week
• scikit-hep / histoprint
  Pretty print histograms to the console
  ☆33Updated this week
• jakirkham / dask-distance
  Distance computations with Dask (akin to scipy.spatial.distance)
  ☆8Updated 6 years ago
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆76Updated this week
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆12Updated 3 months ago