ROCm / pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
☆225Updated this week
Alternatives and similar repositories for pytorch:
Users that are interested in pytorch are comparing it to the libraries listed below
- Next generation BLAS implementation for ROCm platform☆367Updated this week
- Dockerfiles for the various software layers defined in the ROCm software platform☆463Updated 2 weeks ago
- AMD's graph optimization engine.☆216Updated this week
- Stretching GPU performance for GEMMs and tensor contractions.☆237Updated last week
- ROCm Communication Collectives Library (RCCL)☆330Updated this week
- ROCm BLAS marshalling library☆140Updated this week
- A collection of examples for the ROCm software stack☆206Updated this week
- TensorFlow ROCm port☆691Updated this week
- AMD's Machine Intelligence Library☆1,145Updated this week
- ROCm Device Libraries☆97Updated last year
- HIPIFY: Convert CUDA to Portable C++ Code☆574Updated this week
- ROCm Platform Runtime: ROCr a HPC market enhanced HSA based runtime☆247Updated this week
- A GPU benchmark tool for evaluating GPUs and CPUs on mixed operational intensity kernels (CUDA, OpenCL, HIP, SYCL, OpenMP)☆397Updated 3 months ago
- Legacy ROCm Software Platform Documentation☆114Updated last year
- (Deprecated) hipCaffe: the HIP port of Caffe☆124Updated last year
- MIOpenGEMM is now deprecated☆62Updated last year
- RAND library for HIP programming language☆118Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆390Updated this week
- TensorFlow-nGraph bridge☆136Updated 4 years ago
- ROCm Parallel Primitives☆171Updated last week
- ☆251Updated this week
- Examples for HIP☆205Updated 5 months ago
- ROC profiler library. Profiling with perf-counters and derived metrics.☆144Updated last week
- BLAS-like Library Instantiation Software Framework☆139Updated last month
- The NNEF Tools repository contains tools to generate and consume NNEF documents☆224Updated this week
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆106Updated 3 months ago
- oneAPI Collective Communications Library (oneCCL)☆232Updated last week
- ROCm's Thunk Interface☆89Updated last month
- MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into …☆193Updated last week
- AOMP is an open source Clang/LLVM based compiler with added support for the OpenMP® API on Radeon™ GPUs. Use this repository for releas…☆219Updated this week