Skip to content

@axonn-ai AxoNN

Change the repository type filter

All

    Repositories list

    10 repositories

    • axonn

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

    • improved-diffusion

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

    • nanoGPT

      Public
      Demonstration of AxoNN's easy-mode tensor parallelism on the nanoGPT benchmark
      Python
      •
      MIT License
      •6.7k•0•0•0•Updated Dec 26, 2024Dec 26, 2024

    • 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
      •6.5k•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