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
- ☆30Updated 2 weeks ago
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆82Updated 2 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- ☆21Updated 3 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆70Updated 9 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 5 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆72Updated this week
- ☆60Updated 3 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated last week
- Repository of machine learning benchmarks☆48Updated 2 months ago
- ☆21Updated 11 months ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 4 years ago
- A snappy + easy + pretty TUI debugger for Python.☆69Updated 4 months ago
- Named Tensors for Legible Deep Learning in JAX☆218Updated 3 months ago
- A library that translates Python and NumPy to optimized distributed systems code.☆131Updated 3 years ago
- Evaluation Framework for Probabilistic Programming Languages☆104Updated 2 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆75Updated 7 months ago
- Documentation:☆127Updated 2 years ago
- ☆55Updated 2 months ago
- ☆81Updated this week
- TORCH_TRACE parser for PT2☆76Updated this week
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆105Updated this week
- Neural Networks for JAX☆84Updated last year
- PyTorch centric eager mode debugger☆48Updated last year
- Benchmarks to capture important workloads.☆32Updated 2 weeks ago
- Lightning Training strategy for HiveMind☆18Updated 3 weeks ago