axonn-ai/axonn external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

axonn-ai/axonn

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/axonn-ai/axonn)

Close

axonn-ai / axonnView on GitHubMore

• External links
• Readme badge

Parallel framework for training and fine-tuning deep neural networks

☆72Nov 10, 2025Updated 4 months ago

Alternatives and similar repositories for axonn

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

• parallelcodefoundry / HPC-Coder

  View on GitHub
  Scripts for fine-tuning an HPC Code LLM
  ☆16Jul 19, 2024Updated last year
• hpcgroup / AriesNCL

  View on GitHub
  Aries Network Performance Counters Monitoring Library
  ☆11Nov 19, 2020Updated 5 years ago
• hpcgroup / loki

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

  View on GitHub
  Official repo for Detecting, Explaining, and Mitigating Memorization in Diffusion Models (ICLR 2024)
  ☆78Apr 3, 2024Updated last year
• hpcgroup / damselfly

  View on GitHub
  Damselfly Network Simulator
  ☆10Nov 19, 2020Updated 5 years ago
• mcleish7 / gemstone-scaling-laws

  View on GitHub
  Gemstones: A Model Suite for Multi-Faceted Scaling Laws (NeurIPS 2025)
  ☆33Sep 28, 2025Updated 5 months ago
• hatchet / hatchet

  View on GitHub
  Analyze graph/hierarchical performance data using pandas dataframes
  ☆119Oct 22, 2025Updated 4 months ago
• hpcgroup / chatterbug

  View on GitHub
  A suite of communication proxies for HPC applications
  ☆13Jul 7, 2023Updated 2 years ago
• hamidkazemi22 / CLIPInversion

  View on GitHub
  What do we learn from inverting CLIP models?
  ☆58Mar 6, 2024Updated 2 years ago
• devnkong / GOAT

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

  View on GitHub
  Trace Replay and Network Simulation Framework
  ☆21Apr 14, 2021Updated 4 years ago
• at-aaims / forge

  View on GitHub
  ☆15Apr 21, 2025Updated 10 months ago
• aws-neuron / nki-llama

  View on GitHub
  Project showing how to develop NKI kernels for Llama 3.2 1B inference
  ☆21May 29, 2025Updated 9 months ago
• mlcommons / hpc

  View on GitHub
  Reference implementations of MLPerf™ HPC training benchmarks
  ☆50Feb 25, 2025Updated last year
• axonn-ai / distrib-dl-tutorial

  View on GitHub
  Distributed Deep Learning Tutorial
  ☆21Mar 10, 2026Updated last week
• Deep-Learning-Profiling-Tools / triton-samples

  View on GitHub
  ☆14Mar 8, 2025Updated last year
• mcleish7 / arithmetic

  View on GitHub
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆200May 28, 2024Updated last year
• swiss-ai / MoE

  View on GitHub
  some mixture of experts architecture implementations
  ☆26Mar 22, 2024Updated last year
• coreyjadams / CosmicTagger

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

  View on GitHub
  Graph-indexed Pandas DataFrames for analyzing hierarchical performance data
  ☆35Mar 6, 2026Updated last week
• nasaharvest / street2sat

  View on GitHub
  ☆12Jan 5, 2026Updated 2 months ago
• eddieschoute / dissertation-template-umd

  View on GitHub
  A template for University of Maryland dissertations
  ☆27Sep 10, 2023Updated 2 years ago
• leimao / CUTLASS-Examples

  View on GitHub
  CUTLASS and CuTe Examples
  ☆130Nov 30, 2025Updated 3 months ago
• ParCIS / Ok-Topk

  View on GitHub
  Ok-Topk is a scheme for distributed training with sparse gradients. Ok-Topk integrates a novel sparse allreduce algorithm (less than 6k c…
  ☆27Dec 10, 2022Updated 3 years ago
• khaki3 / ptxas-wrapper

  View on GitHub
  A Symbolic Emulator for Shuffle Synthesis on the NVIDIA PTX Code
  ☆15Mar 19, 2023Updated 3 years ago
• parallelcodefoundry / ParEval

  View on GitHub
  Parallel Code Evaluation Benchmark
  ☆44Nov 4, 2025Updated 4 months ago
• IaroslavElistratov / triton-autodiff

  View on GitHub
  ☆18Nov 11, 2025Updated 4 months ago
• Kernel-Machines / kermac

  View on GitHub
  Pytorch routines for (Ker)nel (Mac)hines
  ☆11Oct 10, 2025Updated 5 months ago
• adoxa / errout

  View on GitHub
  Send stderr to stdout, maintaining order.
  ☆11Dec 14, 2013Updated 12 years ago
• CRobeck / instrument-amdgpu-kernels

  View on GitHub
  LLVM/MLIR based compiler instrumentation of AMD GPU kernels
  ☆20Jul 13, 2025Updated 8 months ago
• AMD-AGI / torchtitan-amd

  View on GitHub
  A PyTorch native platform for training generative AI models
  ☆15Nov 18, 2025Updated 4 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
• pnnl / memgaze

  View on GitHub
  ☆17Nov 3, 2025Updated 4 months ago
• aws-samples / end-2-end-3d-ml

  View on GitHub
  This repository features Amazon SageMaker Ground Truth and explains how to ingest raw 3D point cloud data, label it, train a 3D object de…
  ☆13Jun 23, 2022Updated 3 years ago
• meta-pytorch / torchft

  View on GitHub
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆487Updated this week
• UmerHA / triton_util

  View on GitHub
  Make triton easier
  ☆50Jun 12, 2024Updated last year
• vasusingla / simple-data-attribution

  View on GitHub
  A simple and efficient baseline for data attribution
  ☆11Nov 10, 2023Updated 2 years ago
• dayal-kalra / low-memory-adam

  View on GitHub
  ☆15Mar 2, 2025Updated last year
• Jaykef / Triton-nanoGPT

  View on GitHub
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Oct 21, 2024Updated last year