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.☆329Updated this week
- ☆28Updated last year
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 4 years ago
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- ☆30Updated 2 weeks ago
- ☆55Updated last year
- Lightning Training strategy for HiveMind☆18Updated 3 weeks ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 5 months ago
- Repository of machine learning benchmarks☆48Updated 2 months ago
- ☆21Updated 3 years ago
- Test suite for Python array API standard compliance☆70Updated last week
- Productionize machine learning predictions, with ONNX or without☆66Updated 2 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆70Updated 9 months ago
- ☆60Updated 3 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆75Updated 7 months ago
- A snappy + easy + pretty TUI debugger for Python.☆69Updated 4 months ago
- ☆81Updated this week
- PyTorch centric eager mode debugger☆48Updated last year
- ☆55Updated 2 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆47Updated 5 years ago
- Benchmarks to capture important workloads.☆32Updated this week
- Experiment of using Tangent to autodiff triton☆82Updated 2 years ago
- A library that translates Python and NumPy to optimized distributed systems code.☆131Updated 3 years ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆72Updated this week
- Jax SSM Library☆48Updated 3 years ago
- Machine Learning Agility (MLAgility) benchmark and benchmarking tools☆40Updated 6 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆131Updated this week
- Memory Optimizations for Deep Learning (ICML 2023)☆115Updated last year