PyTorch support for Cerebras Wafer Scale Cluster and ML research
Project description
cerebras_pytorch
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.
- Website: https://cerebras.net/
- Documentation: https://docs.cerebras.net/
- Support: support@cerebras.net
- Discord: https://discord.gg/ZqvYS2e2rY
Installation
pip install cerebras_pytorch
Project details
Release history Release notifications | RSS feed
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
- Download URL: cerebras_pytorch-2.3.0-cp38-cp38-manylinux2014_x86_64.whl
- Upload date:
- Size: 40.9 MB
- Tags: CPython 3.8
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.0
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 03f11eacf9387c037c8d4b9d18029f1972bb5c4043ba8ad1e5aff72ee2ca513c |
|
MD5 | 5f1a89af2070edc052b908b27410b886 |
|
BLAKE2b-256 | f5eb517308b8bae3c55ff4b9597d5d589493d4d192dbc54a7f63477efdc53fc3 |