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
cerebras.modelzoo.tools.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17
next
cerebras.modelzoo.tools.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17