Quansight-Labs / numpy_pytorch_interop
☆20Updated 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
Sorting:
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆60Updated last month
- ☆27Updated 4 months ago
- ☆51Updated 9 months ago
- Experimental plugin for scikit-learn to be able to run (some estimators) on Intel GPUs via numba-dpex.☆16Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ONNX-backed array library that is compliant with the Array API standard.☆55Updated this week
- ArrayViews: creating specific views to array storage objects☆16Updated 6 years ago
- A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.☆12Updated last year
- An Aspiring Drop-In Replacement for Pandas at Scale☆75Updated 3 years ago
- TritonParse is a new tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source …☆14Updated last week
- PyTorch centric eager mode debugger☆47Updated 5 months ago
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- Notes and artifacts from the ONNX steering committee☆26Updated this week
- Data and tooling to compare the API surfaces of various array libraries.☆54Updated 3 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- ☆21Updated 2 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 11 months ago
- Tools for JAX☆46Updated this week
- Bayesian optimization of discrete sequences☆21Updated 3 years ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Jax SSM Library☆49Updated 2 years ago
- ☆45Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆111Updated last week
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- A conda-smithy repository for llvmdev.☆9Updated this week
- Better bindings for Python☆17Updated 2 years ago
- Benchmark structured generation libraries☆27Updated 6 months ago
- ☆60Updated 3 years ago
- ☆47Updated last month
- A sample pattern for running CI tests on Modal☆17Updated last month