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
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
Close
Hashes for cerebras_pytorch-2.2.0-cp38-cp38-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f0bff6dfa644a1c872143b11fbad2ecdf32ffa0fc54bd6dbc0d64c11ce7a88da |
|
MD5 | 404d3b7e48e7fabe0a8d38699e783650 |
|
BLAKE2b-256 | eb9df1c354bf988b85fad785e464e27b80571437e919ad79ef5d71d54b31043d |