Clone Cerebras Model Zoo#

Cerebras Model Zoo repository contains examples of common deep learning models that can be trained on Cerebras cluster.

This repository can be used for:

  • Getting started. It contains:

    • Reference implementations in PyTorch and TensorFlow of common deep learning models

    • Data preparation scripts to use your own data

    • Configuration files for multiple model sizes

  • Benchmarking. It contains tuned implementations with optimal performance in Cerebras hardware.

  • Developing new models. It contains:

    • Cerebras-customized PyTorch and Tensorflow Layer APIs that simplify the compilation process.

    • Examples that demonstrate best practices for coding your own model.

Note

The Cerebras Model Zoo for the Cerebras Original Installation is available at https://github.com/Cerebras/modelzoo/tree/original_cerebras_installation. A list of supported models can be found here.

To clone the Cerebras Model Zoo Repository in your preferred directory, use:

git clone -b original_cerebras_installation https://github.com/Cerebras/modelzoo.git

Warning

Only models in pipelined execution are fully supported in the Original Cerebras Installation