openxla / xprofLinks
A profiling and performance analysis tool for machine learning
☆460Updated this week
Alternatives and similar repositories for xprof
Users that are interested in xprof are comparing it to the libraries listed below
Sorting:
- A performant and modular runtime for TensorFlow☆756Updated 4 months ago
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆735Updated last month
- PyTorch RFCs (experimental)☆136Updated 7 months ago
- Continuous builder and binary build scripts for pytorch☆356Updated 4 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆282Updated 3 weeks ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆411Updated this week
- Guide for building custom op for TensorFlow☆385Updated 2 years ago
- TensorFlow Estimator☆302Updated last year
- ☆423Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,071Updated last year
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆368Updated this week
- The Triton backend for the PyTorch TorchScript models.☆170Updated this week
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,512Updated this week
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64Updated this week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆913Updated this week
- Build-related tools for TensorFlow☆301Updated last month
- Computation using data flow graphs for scalable machine learning☆68Updated this week
- Example code and applications for machine learning on Graphcore IPUs☆333Updated last year
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆104Updated this week
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆46Updated 2 years ago
- A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.☆1,559Updated last week
- TensorFlow-nGraph bridge☆136Updated 4 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆1,003Updated last week
- Utilities for passing TensorFlow-related metadata between tools☆110Updated 5 months ago
- A GPU performance profiling tool for PyTorch models☆509Updated 4 years ago
- PyTorch interface for the IPU☆181Updated 2 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆162Updated 2 weeks ago
- Providing reproducibility in deep learning frameworks☆434Updated last year
- For recording and retrieving metadata associated with ML developer and data scientist workflows.☆667Updated 9 months ago
- Python bindings for NVTX☆67Updated 2 years ago