Introduction
Get Started
Prepare your data
Cerebras PyTorch API
cerebras_pytorch
cerebras_pytorch.amp
cerebras_pytorch.optim
cerebras_pytorch.sparse
cerebras_pytorch.metrics
Cerebras Model Zoo
Cerebras Guides
Fundamentals
Support
Classes
ConfigConverter_GPT_Neox_HF_CS17
ConfigConverter_GPT_Neox_HF_CS18
ConfigConverter_GPT_Neox_HF_CS20
ConfigConverter_GPT_Neox_HF_CS21
ConfigConverter_GPT_Neox_Headless_CS18_CS20
NeoX uses the GPTJ backbone
Converter_GPT_Neox_Attention_HF_CS17
Converter_GPT_Neox_Headless_HF_CS17
Converter_GPT_Neox_Headless_HF_CS18
Converter_GPT_Neox_Headless_HF_CS20
Converter_GPT_Neox_Headless_HF_CS21
Converter_GPT_Neox_LMHeadModel_CS18_CS20
Converter_GPT_Neox_LMHeadModel_CS20_CS21
Converter_GPT_Neox_LMHeadModel_HF_CS17
Converter_GPT_Neox_LMHeadModel_HF_CS18
Converter_GPT_Neox_LMHeadModel_HF_CS20
Converter_GPT_Neox_LMHeadModel_HF_CS21
previous
modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17
next
modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17