Provides a PyTorch backend that runs PyTorch models and operations on Google's Tensor Processing Units (TPUs).
Python library providing GPU-accelerated tensor computation like NumPy, plus a tape-based autograd system for building deep neural networks.
PyTorch library for audio generation research, providing inference and training code for MusicGen, AudioGen, and EnCodec.
Python toolkit for speaker diarization built on PyTorch, with pretrained pipelines and models that can be fine-tuned on your own data.
Open-source large language models and tooling for financial tasks such as sentiment analysis, forecasting, and robo-advisory.