Skip to main content

PyTorch support for Cerebras Wafer Scale Cluster and ML research

Project description

cerebras_pytorch

PyPI Version PyPI - Python Version Discord PyPI - Downloads

The cerebras_pytorch package is a component of Cerebras's Wafer-Scale cluster software stack, offering a framework for training and evaluating neural networks with near-perfect linear scaling across millions of cores without the inconvenience of distributed computing. Additionally, it provides general ML helpers for running on CPU/GPU, empowering Cerebras' modelzoo.

Installation

pip install cerebras_pytorch

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

File details

Details for the file cerebras_pytorch-2.3.0-cp38-cp38-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for cerebras_pytorch-2.3.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 03f11eacf9387c037c8d4b9d18029f1972bb5c4043ba8ad1e5aff72ee2ca513c
MD5 5f1a89af2070edc052b908b27410b886
BLAKE2b-256 f5eb517308b8bae3c55ff4b9597d5d589493d4d192dbc54a7f63477efdc53fc3

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page