Quansight-Labs / numpy_pytorch_interop

Related projects ⓘ

Alternatives and complementary repositories for numpy_pytorch_interop

• modal-labs / ci-on-modal
  A sample pattern for running CI tests on Modal
  ☆13Updated 2 months ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆22Updated last week
• iree-org / iree-jax
  ☆48Updated 3 months ago
• distdl / distdl
  ☆21Updated 2 years ago
• soda-inria / sklearn-numba-dpex
  Experimental plugin for scikit-learn to be able to run (some estimators) on Intel GPUs via numba-dpex.
  ☆15Updated 8 months ago
• uber / hypothesis-gufunc
  Extension to hypothesis for testing numpy general universal functions
  ☆39Updated 3 years ago
• data-apis / array-api-tests
  Test suite for the PyData Array APIs standard
  ☆65Updated this week
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆16Updated 4 months ago
• jorenham / jax_pep503
  PEP 503 repository index for jax[cuda]
  ☆22Updated this week
• zdevito / custom_loader
  ☆12Updated 3 years ago
• colesbury / nogil-wheels
  GitHub actions to build wheels for nogil Python
  ☆14Updated 7 months ago
• NVIDIA / numba-cuda
  ☆36Updated last week
• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated this week
• andyljones / modulepickle
  Extends cloudpickle to dynamically pickle modules from your working directory
  ☆10Updated 5 years ago
• data-apis / array-api-compat
  Compatibility layer for common array libraries to support the Array API
  ☆79Updated this week
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆30Updated last year
• jax-ml / jax-ai-stack
  ☆77Updated this week
• probml / JSL
  Jax SSM Library
  ☆49Updated last year
• suo / lintrunner
  ☆24Updated 6 months ago
• 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 / init2winit
  ☆72Updated last week
• rsokl / phantom-tensors
  Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing
  ☆17Updated 3 weeks ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• PierrickPochelu / JaxDecompiler
  Jax Decompiler
  ☆13Updated 3 months ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• dottxt-ai / benchmarks
  Benchmark structured generation libraries
  ☆22Updated 3 weeks ago
• onnx / steering-committee
  Notes and artifacts from the ONNX steering committee
  ☆25Updated last week
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated last month
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated last year