Alternatives and similar repositories for axonn

Users that are interested in axonn are comparing it to the libraries listed below

• hpcgroup / pipit

  View on GitHub
  Analyze parallel execution traces using pandas dataframes
  ☆24Oct 22, 2025Updated 3 months ago
• coreyjadams / CosmicTagger

  View on GitHub
  Cosmic Tagging Network for Neutrino Physics
  ☆13Jun 26, 2024Updated last year
• mlcommons / hpc

  View on GitHub
  Reference implementations of MLPerf™ HPC training benchmarks
  ☆49Feb 25, 2025Updated 11 months ago
• hpcgroup / loki

  View on GitHub
  Algorithms for approximate attention in LLMs
  ☆21Apr 14, 2025Updated 10 months ago
• YuxinWenRick / diffusion_memorization

  View on GitHub
  Official repo for Detecting, Explaining, and Mitigating Memorization in Diffusion Models (ICLR 2024)
  ☆77Apr 3, 2024Updated last year
• argonne-lcf / alcl

  View on GitHub
  Argonne Leadership Computing Facility OpenCL tutorial
  ☆10Aug 22, 2025Updated 5 months ago
• NERSC / intro-HPC-bootcamp-2023

  View on GitHub
  ☆14Sep 7, 2023Updated 2 years ago
• Deep-Learning-Profiling-Tools / triton-samples

  View on GitHub
  ☆14Mar 8, 2025Updated 11 months ago
• hpcgroup / AriesNCL

  View on GitHub
  Aries Network Performance Counters Monitoring Library
  ☆11Nov 19, 2020Updated 5 years ago
• HPCToolkit / hpctoolkit-tutorial-examples

  View on GitHub
  CPU and GPU tutorial examples
  ☆13Apr 4, 2025Updated 10 months ago
• sparticlesteve / cosmoflow-benchmark

  View on GitHub
  Benchmark implementation of CosmoFlow in TensorFlow Keras
  ☆22Feb 7, 2024Updated 2 years ago
• axonn-ai / distrib-dl-tutorial

  View on GitHub
  Distributed Deep Learning Tutorial
  ☆16Nov 17, 2025Updated 2 months ago
• hpcgroup / damselfly

  View on GitHub
  Damselfly Network Simulator
  ☆10Nov 19, 2020Updated 5 years ago
• AMD-AGI / torchtitan-amd

  View on GitHub
  A PyTorch native platform for training generative AI models
  ☆15Nov 18, 2025Updated 2 months ago
• IaroslavElistratov / triton-autodiff

  View on GitHub
  ☆18Nov 11, 2025Updated 3 months ago
• codeplaysoftware / cuda-to-sycl-nbody

  View on GitHub
  ☆14Mar 1, 2025Updated 11 months ago
• hpdps-group / ICS23-GPULZ

  View on GitHub
  GPULZ: Optimizing LZSS Lossless Compression for Multi-byte Data on Modern GPUs
  ☆16Apr 18, 2025Updated 9 months ago
• aws-neuron / nki-library

  View on GitHub
  ☆42Jan 24, 2026Updated 3 weeks ago
• uclasystem / MemLiner

  View on GitHub
  MemLiner is a remote-memory-friendly runtime system.
  ☆31Nov 1, 2022Updated 3 years ago
• llnl / hatchet

  View on GitHub
  Graph-indexed Pandas DataFrames for analyzing hierarchical performance data
  ☆34Jan 30, 2026Updated 2 weeks ago
• hamidkazemi22 / CLIPInversion

  View on GitHub
  What do we learn from inverting CLIP models?
  ☆58Mar 6, 2024Updated last year
• NMSU-PEARL / GPUs-Energy

  View on GitHub
  [CF ’20] Verified Instruction-Level Energy Consumption Measurement for NVIDIA GPUs
  ☆15Dec 11, 2020Updated 5 years ago
• NERSC / nersc-dl-wandb

  View on GitHub
  Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines
  ☆13Aug 7, 2023Updated 2 years ago
• at-aaims / OpenMxP

  View on GitHub
  This is the open source version of HPL-MXP. The code performance has been verified on Frontier
  ☆18Jul 9, 2025Updated 7 months ago
• ROCm / aws-ofi-rccl

  View on GitHub
  ☆17Nov 11, 2025Updated 3 months ago
• hpcgroup / chatterbug

  View on GitHub
  A suite of communication proxies for HPC applications
  ☆13Jul 7, 2023Updated 2 years ago
• CrossFacilityWorkflows / DOE-HPC-workflow-training

  View on GitHub
  Repo for a DOE HPC workflow training event
  ☆13Apr 28, 2023Updated 2 years ago
• swiss-ai / MoE

  View on GitHub
  some mixture of experts architecture implementations
  ☆25Mar 22, 2024Updated last year
• parallelcodefoundry / ParEval

  View on GitHub
  Parallel Code Evaluation Benchmark
  ☆42Nov 4, 2025Updated 3 months ago
• meta-pytorch / torchft

  View on GitHub
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆476Feb 3, 2026Updated 2 weeks ago
• CRobeck / instrument-amdgpu-kernels

  View on GitHub
  LLVM/MLIR based compiler instrumentation of AMD GPU kernels
  ☆20Jul 13, 2025Updated 7 months ago
• argonne-lcf / Megatron-DeepSpeed

  View on GitHub
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆17Updated this week
• graphcore-research / unit-scaling-demo

  View on GitHub
  Unit Scaling demo and experimentation code
  ☆16Mar 12, 2024Updated last year
• devnkong / GOAT

  View on GitHub
  Official implementation of GOAT model (ICML2023)
  ☆38Jul 3, 2023Updated 2 years ago
• mingfeima / pytorch_profiler_parser

  View on GitHub
  parser script to process pytorch autograd profiler result, convert json file to excel.
  ☆14Oct 8, 2019Updated 6 years ago
• argonne-lcf / copper

  View on GitHub
  scalable data movement in Exascale Supercomputers
  ☆17Dec 4, 2025Updated 2 months ago
• FindHao / drgpu

  View on GitHub
  A Top-Down Profiler for GPU Applications
  ☆22Feb 29, 2024Updated last year
• lanl / benchmarks

  View on GitHub
  Benchmarks
  ☆17Apr 28, 2025Updated 9 months ago
• srush / anynp

  View on GitHub
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Jun 18, 2024Updated last year