Metadata-Version: 2.1
Name: cerebras-pytorch
Version: 2.8.0
Summary: PyTorch support for Cerebras Wafer Scale Cluster and ML research
Home-page: https://cerebras.net/
Author: Cerebras Systems
Author-email: support@cerebras.net
Classifier: Development Status :: 5 - Production/Stable
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: BSD License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: System :: Hardware
Classifier: Programming Language :: Python :: 3
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: pip (>=21.1.2)
Requires-Dist: numpy (==1.25.0)
Requires-Dist: pandas (==2.2.3)
Requires-Dist: protobuf (==4.21.12)
Requires-Dist: grpcio (==1.51.1)
Requires-Dist: grpcio-tools (==1.51.1)
Requires-Dist: multiprocess (==0.70.16)
Requires-Dist: boto3
Requires-Dist: dill (==0.3.8)
Requires-Dist: tblib (==3.0.0)
Requires-Dist: tqdm (>=4.42.1)
Requires-Dist: psutil (>=5.8.0)
Requires-Dist: urllib3 (<2)
Requires-Dist: tabulate (==0.9.0)
Requires-Dist: tensorboard (==2.17.1)
Requires-Dist: torch (==2.4.0)
Requires-Dist: cerebras-appliance (==2.8.0)

# cerebras_pytorch

[![PyPI Version](https://img.shields.io/pypi/v/cerebras_pytorch)](https://pypi.org/project/cerebras_pytorch/)
[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/cerebras_pytorch)](https://pypi.org/project/cerebras_pytorch/)
[![Discord](https://img.shields.io/discord/1085960591052644463)](https://discord.gg/ZqvYS2e2rY)
[![PyPI - Downloads](https://img.shields.io/pypi/dm/cerebras_pytorch)](https://pypi.org/project/cerebras_pytorch/#files)

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](https://github.com/Cerebras/modelzoo).

* **Website**: https://cerebras.net/
* **Documentation**: https://docs.cerebras.net/
* **Support**: support@cerebras.net
* **Discord**: https://discord.gg/ZqvYS2e2rY

## Installation

```bash
pip install cerebras_pytorch
```

