Quansight-Labs / numpy_pytorch_interopLinks
☆20Updated 2 years ago
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.☆319Updated this week
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Neural Networks for JAX☆84Updated last year
- Experiment of using Tangent to autodiff triton☆81Updated last year
- ☆60Updated 3 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 3 months ago
- A snappy + easy + pretty TUI debugger for Python.☆70Updated 2 months ago
- ☆54Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated 2 months ago
- ☆21Updated 9 months ago
- ☆28Updated 10 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- TORCH_LOGS parser for PT2☆69Updated last month
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- Test suite for Python array API standard compliance☆69Updated last week
- Repository of machine learning benchmarks☆48Updated 3 weeks ago
- ☆52Updated 3 weeks ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 4 years ago
- ☆12Updated 3 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆61Updated last week
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆74Updated 5 months ago
- Named Tensors for Legible Deep Learning in JAX☆212Updated last month
- A place to store reusable transformer components of my own creation or found on the interwebs☆63Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆64Updated 2 weeks ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆44Updated last year
- ☆29Updated 3 weeks ago
- ☆79Updated last week
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆69Updated 7 months ago
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆103Updated this week
- PyTorch centric eager mode debugger☆48Updated 11 months ago