Quansight-Labs / numpy_pytorch_interopLinks
☆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
- ☆28Updated 4 months ago
- ☆52Updated 9 months ago
- Benchmarking PyTorch 2.0 different models☆21Updated 2 years ago
- ONNX-backed array library that is compliant with the Array API standard.☆57Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆110Updated 3 weeks ago
- Experimental plugin for scikit-learn to be able to run (some estimators) on Intel GPUs via numba-dpex.☆16Updated last year
- Tools for JAX☆47Updated last week
- Data and tooling to compare the API surfaces of various array libraries.☆54Updated 4 months ago
- NumPy ufuncs and utilities.☆52Updated last week
- ☆21Updated 3 months ago
- Neural Networks for JAX☆84Updated 8 months ago
- Better bindings for Python☆17Updated 2 years ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- PyTorch centric eager mode debugger☆47Updated 5 months ago
- Benchmarks to capture important workloads.☆31Updated 4 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆44Updated 10 months ago
- ☆47Updated last week
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 3 years ago
- Build modern UIs in Jupyter with Python☆12Updated 2 years ago
- Personal solutions to the Triton Puzzles☆18Updated 10 months ago
- ☆21Updated 2 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆268Updated last week
- TORCH_LOGS parser for PT2☆38Updated last week
- Efficient in-memory representation for ONNX, in Python☆13Updated this week
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated last year
- Compatibility layer for common array libraries to support the Array API☆97Updated 2 weeks ago
- ☆18Updated this week
- Collection of scripts to build PyTorch and the domain libraries from source.☆11Updated this week