Skip to content

@axonn-ai AxoNN

Change the repository type filter

All

    Repositories list

    11 repositories

    • sc2025-pccl-reproducer

      Public
      Communication library for Deep Learning
      Python
      •0•1•0•0•Updated Apr 29, 2025Apr 29, 2025

    • nanoGPT

      Public
      Demonstration of AxoNN's easy-mode tensor parallelism on the nanoGPT benchmark
      Python
      •
      MIT License
      •6.8k•0•0•0•Updated Apr 23, 2025Apr 23, 2025

    • axonn

      Public
      A parallel framework for training deep neural networks
      parallel-computingneural-networksmachine-learningdeep-learning
      Python
      •
      Apache License 2.0
      •5•58•9•3•Updated Mar 16, 2025Mar 16, 2025

    • improved-diffusion

      Public
      Release for Improved Denoising Diffusion Probabilistic Models
      Python
      •
      MIT License
      •507•0•0•3•Updated Jan 27, 2025Jan 27, 2025

    • distrib-dl-tutorial

      Public
      Python
      •1•10•0•0•Updated Nov 18, 2024Nov 18, 2024

    • vllm

      Public
      A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      •
      Apache License 2.0
      •7.3k•0•0•0•Updated Nov 6, 2024Nov 6, 2024

    • litgpt

      Public
      20+ high-performance LLM implementations with recipes to pretrain, finetune and deploy at scale.
      Python
      •
      Apache License 2.0
      •1.2k•0•0•1•Updated Sep 25, 2024Sep 25, 2024

    • axonn-examples

      Public
      Python
      •0•1•0•1•Updated Sep 13, 2024Sep 13, 2024

    • Megatron-AxoNN

      Public
      A GPT benchmark with AxoNN's model parallelism
      Python
      •
      Other
      •2.7k•1•0•3•Updated Aug 8, 2024Aug 8, 2024

    • getting-started

      Public
      Scripts for getting setup on various PSSG clusters
      Shell
      •0•0•0•0•Updated Jun 5, 2024Jun 5, 2024

    • accelerate

      Public
      🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
      Python
      •
      Apache License 2.0
      •1.1k•3•0•0•Updated Feb 28, 2024Feb 28, 2024