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

If you're not sure about the file name format, learn more about wheel file names.

cerebras_pytorch-2.6.0-cp311-cp311-manylinux2014_x86_64.whl (43.7 MB view details)

Uploaded CPython 3.11

File details

Details for the file cerebras_pytorch-2.6.0-cp311-cp311-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for cerebras_pytorch-2.6.0-cp311-cp311-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 c484140c0bfe254891526df0b8a6a541ddf3f7005b6d4724e87ca6f4f336e2b2
MD5 8ab36fe7596240017120002527438f47
BLAKE2b-256 a4694f276181cf7eaa5573b5871a0cbe7e913106a6a9ab354618b9d33b9a1b68

See more details on using hashes here.

Supported by

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