graphcore / poptorch
PyTorch interface for the IPU
☆179Updated last year
Alternatives and similar repositories for poptorch:
Users that are interested in poptorch are comparing it to the libraries listed below
- Poplar Advanced Runtime for the IPU☆6Updated last year
- TensorFlow for the IPU☆78Updated last year
- Example code and applications for machine learning on Graphcore IPUs☆320Updated last year
- Poplar libraries☆118Updated last year
- Training material for IPU users: tutorials, feature examples, simple applications☆86Updated 2 years ago
- Graph algorithms for machine learning frameworks☆29Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆256Updated 2 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated 4 months ago
- Python bindings for NVTX☆66Updated last year
- PyTorch RFCs (experimental)☆131Updated 7 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- Customized matrix multiplication kernels☆54Updated 3 years ago
- ☆50Updated last year
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆156Updated 4 months ago
- Implementation of a Transformer, but completely in Triton☆263Updated 3 years ago
- ☆39Updated 4 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆389Updated this week
- A library of GPU kernels for sparse matrix operations.☆262Updated 4 years ago
- A library to analyze PyTorch traces.☆366Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆109Updated this week
- This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.☆38Updated last year
- TensorFlow-nGraph bridge☆136Updated 4 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆51Updated 7 years ago
- ☆163Updated 10 months ago
- ☆186Updated this week
- Torch Distributed Experimental☆115Updated 8 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆314Updated this week
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago