DeepSpeed

DeepSpeed

Microsoft
+
+

Related Products

  • Pipedrive
    10,536 Ratings
    Visit Website
  • Checksum.ai
    1 Rating
    Visit Website
  • Gemini Enterprise Agent Platform
    999 Ratings
    Visit Website
  • LM-Kit.NET
    29 Ratings
    Visit Website
  • Runpod
    230 Ratings
    Visit Website
  • Google AI Studio
    40 Ratings
    Visit Website
  • LTX
    182 Ratings
    Visit Website
  • StackAI
    54 Ratings
    Visit Website
  • Codemagic
    138 Ratings
    Visit Website
  • KrakenD
    71 Ratings
    Visit Website

About

​Axolotl is an open source tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures. It enables users to train models, supporting methods like full fine-tuning, LoRA, QLoRA, ReLoRA, and GPTQ. Users can customize configurations using simple YAML files or command-line interface overrides, and load different dataset formats, including custom or pre-tokenized datasets. Axolotl integrates with technologies like xFormers, Flash Attention, Liger kernel, RoPE scaling, and multipacking, and works with single or multiple GPUs via Fully Sharded Data Parallel (FSDP) or DeepSpeed. It can be run locally or on the cloud using Docker and supports logging results and checkpoints to several platforms. It is designed to make fine-tuning AI models friendly, fast, and fun, without sacrificing functionality or scale.

About

DeepSpeed is an open source deep learning optimization library for PyTorch. It's designed to reduce computing power and memory use, and to train large distributed models with better parallelism on existing computer hardware. DeepSpeed is optimized for low latency, high throughput training. DeepSpeed can train DL models with over a hundred billion parameters on the current generation of GPU clusters. It can also train up to 13 billion parameters in a single GPU. DeepSpeed is developed by Microsoft and aims to offer distributed training for large-scale models. It's built on top of PyTorch, which specializes in data parallelism.

Platforms Supported

Windows Not Supported
Mac Not Supported
Linux Not Supported
Cloud Supported
On-Premises Not Supported
iPhone Not Supported
iPad Not Supported
Android Not Supported
Chromebook Not Supported

Platforms Supported

Windows Supported
Mac Supported
Linux Supported
Cloud Supported
On-Premises Supported
iPhone Not Supported
iPad Not Supported
Android Not Supported
Chromebook Not Supported

Audience

Data scientists seeking a solution to fine-tune large language models across various architectures

Audience

Deep learning model developers

Support

Phone Support Not Supported
24/7 Live Support Not Supported
Online Supported

Support

Phone Support Not Supported
24/7 Live Support Not Supported
Online Not Supported

API

Offers API Not Supported

API

Offers API Not Supported

Screenshots and Videos

Screenshots and Videos

Pricing

Free
Free Version Supported
Free Trial Not Supported

Pricing

Free
Open source
Free Version Supported
Free Trial Not Supported

Reviews/Ratings

Overall 0.0 / 5
ease 0.0 / 5
features 0.0 / 5
design 0.0 / 5
support 0.0 / 5

This software hasn't been reviewed yet. Be the first to provide a review:

Review this Software

Reviews/Ratings

Overall 0.0 / 5
ease 0.0 / 5
features 0.0 / 5
design 0.0 / 5
support 0.0 / 5

This software hasn't been reviewed yet. Be the first to provide a review:

Review this Software

Training

Documentation Supported
Webinars Not Supported
Live Online Not Supported
In Person Not Supported

Training

Documentation Supported
Webinars Not Supported
Live Online Not Supported
In Person Not Supported

Company Information

Axolotl
United States
axolotl.ai/

Company Information

Microsoft
Founded: 1975
United States
www.deepspeed.ai/

Alternatives

LLaMA-Factory

LLaMA-Factory

hoshi-hiyouga

Alternatives

GPT-NeoX

GPT-NeoX

EleutherAI
AWS Neuron

AWS Neuron

Amazon Web Services

Categories

AI Fine-Tuning Supported

Categories

AI Development Supported
Deep Learning Supported

Integrations

Axolotl Not Supported
Cerebras Supported
Comet LLM Not Supported
DeepSpeed Supported
Docker Supported
Falcon Supported
GPT-J Supported
Hugging Face Supported
Latitude Supported
Llama Supported
MLflow Supported
MPT-7B Supported
Mistral AI Supported
Modal Supported
OpenPipe Supported
PyTorch Not Supported
Pythia Supported
Runpod Supported
TensorWave Supported
XGen Security Supported

Integrations

Axolotl Supported
Cerebras Not Supported
Comet LLM Supported
DeepSpeed Not Supported
Docker Not Supported
Falcon Not Supported
GPT-J Not Supported
Hugging Face Not Supported
Latitude Not Supported
Llama Not Supported
MLflow Not Supported
MPT-7B Not Supported
Mistral AI Not Supported
Modal Not Supported
OpenPipe Not Supported
PyTorch Supported
Pythia Not Supported
Runpod Not Supported
TensorWave Not Supported
XGen Security Not Supported
Claim Axolotl and update features and information
Claim Axolotl and update features and information
Claim DeepSpeed and update features and information
Claim DeepSpeed and update features and information