DeepSpeed

DeepSpeed

Microsoft
+
+

Related Products

  • Pipedrive
    10,456 Ratings
    Visit Website
  • Checksum.ai
    1 Rating
    Visit Website
  • Gemini Enterprise Agent Platform
    984 Ratings
    Visit Website
  • LM-Kit.NET
    29 Ratings
    Visit Website
  • Runpod
    220 Ratings
    Visit Website
  • Google AI Studio
    30 Ratings
    Visit Website
  • StackAI
    53 Ratings
    Visit Website
  • Codemagic
    137 Ratings
    Visit Website
  • KrakenD
    71 Ratings
    Visit Website
  • AdRem NetCrunch
    158 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
Mac
Linux
Cloud
On-Premises
iPhone
iPad
Android
Chromebook

Platforms Supported

Windows
Mac
Linux
Cloud
On-Premises
iPhone
iPad
Android
Chromebook

Audience

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

Audience

Deep learning model developers

Support

Phone Support
24/7 Live Support
Online

Support

Phone Support
24/7 Live Support
Online

API

Offers API

API

Offers API

Screenshots and Videos

Screenshots and Videos

Pricing

Free
Free Version
Free Trial

Pricing

Free
Open source
Free Version
Free Trial

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
Webinars
Live Online
In Person

Training

Documentation
Webinars
Live Online
In Person

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

Categories

Integrations

Axolotl
Cake AI
Cerebras
Comet
Comet LLM
Docker
Falcon
Gemma
Llama
MPT-7B
Mistral AI
Modal
Nurix
PyTorch
Pythia
Python
Qwen
Runpod
Weights & Biases
XGen Security

Integrations

Axolotl
Cake AI
Cerebras
Comet
Comet LLM
Docker
Falcon
Gemma
Llama
MPT-7B
Mistral AI
Modal
Nurix
PyTorch
Pythia
Python
Qwen
Runpod
Weights & Biases
XGen Security
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