Quansight-Labs / numpy_pytorch_interop
☆19Updated last year
Alternatives and similar repositories for numpy_pytorch_interop:
Users that are interested in numpy_pytorch_interop are comparing it to the libraries listed below
- ☆50Updated 5 months ago
- A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.☆12Updated 10 months ago
- ☆23Updated this week
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- TORCH_LOGS parser for PT2☆30Updated last week
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- Experimental plugin for scikit-learn to be able to run (some estimators) on Intel GPUs via numba-dpex.☆15Updated 10 months ago
- ☆106Updated this week
- ☆21Updated 2 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆238Updated this week
- Benchmarking PyTorch 2.0 different models☆21Updated last year
- A tracing JIT compiler for PyTorch☆12Updated 3 years ago
- ☆73Updated this week
- ☆26Updated last month
- A sample pattern for running CI tests on Modal☆14Updated 3 months ago
- Tensor Parallelism with JAX + Shard Map☆11Updated last year
- This is a port of Mistral-7B model in JAX☆30Updated 6 months ago
- benchmarking some transformer deployments☆26Updated last year
- GitHub actions to build wheels for nogil Python☆14Updated 9 months ago
- Experiment of using Tangent to autodiff triton☆74Updated 11 months ago
- Notes and artifacts from the ONNX steering committee☆25Updated last week
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆84Updated this week
- Data and tooling to compare the API surfaces of various array libraries.☆54Updated 7 months ago
- Named Tensors for Legible Deep Learning in JAX☆159Updated last week
- PyTorch centric eager mode debugger☆43Updated last month
- ☆21Updated 2 years ago
- A compiler written in Mojo 🔥 and generates RISC-V assembly☆14Updated 6 months ago
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- A tracing JIT for PyTorch☆17Updated 2 years ago
- ☆41Updated 3 weeks ago