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:
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆292Updated 2 weeks ago
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- ☆60Updated 3 years ago
- ☆21Updated 3 years ago
- Efficient in-memory representation for ONNX, in Python☆25Updated this week
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 3 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆63Updated 4 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated 3 weeks ago
- TorchFix - a linter for PyTorch-using code with autofix support☆145Updated last week
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- TORCH_LOGS parser for PT2☆57Updated this week
- ☆21Updated 5 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆59Updated this week
- Repository of machine learning benchmarks☆40Updated last week
- ☆52Updated last year
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAX☆201Updated last week
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated last month
- ☆28Updated 7 months ago
- PyTorch centric eager mode debugger☆48Updated 8 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- Neural Networks for JAX☆84Updated 11 months ago
- ☆50Updated 3 months ago
- Productionize machine learning predictions, with ONNX or without☆66Updated last year
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆97Updated this week
- ☆28Updated last month
- Test suite for Python array API standard compliance☆68Updated 2 weeks ago
- ☆77Updated last week