DeepSpeed

DeepSpeed

Microsoft
+
+

Related Products

  • Ango Hub
    15 Ratings
    Visit Website
  • Vertex AI
    783 Ratings
    Visit Website
  • Pipedrive
    9,644 Ratings
    Visit Website
  • RunPod
    205 Ratings
    Visit Website
  • Google AI Studio
    11 Ratings
    Visit Website
  • LM-Kit.NET
    23 Ratings
    Visit Website
  • StackAI
    48 Ratings
    Visit Website
  • OORT DataHub
    13 Ratings
    Visit Website
  • KrakenD
    71 Ratings
    Visit Website
  • AdRem NetCrunch
    148 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
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

Cerebras
Comet LLM
DeepSpeed
Falcon
GPT-J
Latitude
Llama
MLflow
MPT-7B
Modal
Nurix
OpenPipe
Phi-2
PyTorch
Pythia
Python
Qwen
TensorWave
Weights & Biases
XGen Security

Integrations

Cerebras
Comet LLM
DeepSpeed
Falcon
GPT-J
Latitude
Llama
MLflow
MPT-7B
Modal
Nurix
OpenPipe
Phi-2
PyTorch
Pythia
Python
Qwen
TensorWave
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