Alternatives and similar repositories for nsight-python

Users that are interested in nsight-python are comparing it to the libraries listed below

• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 8 months ago
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆124Updated 2 weeks ago
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆61Updated last week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆151Updated 2 years ago
• google / aqt
  ☆337Updated last week
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆640Updated this week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆312Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆454Updated 2 weeks ago
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆78Updated 2 months ago
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆170Updated this week
• lianakoleva / no-libtorch-compile
  ☆21Updated 8 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated this week
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆57Updated last week
• dropbox / gemlite
  Fast low-bit matmul kernels in Triton
  ☆401Updated last week
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆226Updated last year
• daniel-geon-park / triton_bwd
  Automatic differentiation for Triton Kernels
  ☆30Updated 3 months ago
• albanD / subclass_zoo
  ☆184Updated last year
• Jokeren / triton-samples
  ☆28Updated 10 months ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆103Updated this week
• google / praxis
  ☆190Updated last week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆256Updated this week
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆307Updated 3 months ago
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆136Updated 6 months ago
• open-lm-engine / accelerated-model-architectures
  A bunch of kernels that might make stuff slower 😉
  ☆65Updated last week
• google / saxml
  ☆147Updated 3 weeks ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 4 months ago
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆46Updated last year