11 active products

PyTorch

Publisher of Captum and Bayesian Optimization Library (botorch)

Consolidated Product Portfolio

Vision
Vision

Datasets, Transforms and Models specific to Computer Vision

Machine Learning Model TrainingOpen Source
TensorRT
TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Machine Learning Model TrainingOpen Source
PyTorch
PyTorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Machine Learning Model TrainingOpen Source
Bayesian Optimization Library (botorch)

BoTorch is a library for Bayesian optimization built on PyTorch.

Devcy HypertuneCommercial
Ao
Ao

PyTorch native quantization for training and inference

Model DeploymentOpen Source
XLA
XLA

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Machine Learning Model TrainingOpen Source
Ignite
Ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Machine Learning Model TrainingOpen Source
Audio
Audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

Machine Learning Model TrainingOpen Source
Captum

Captum is a library for model interpretability in PyTorch.

Ai EvaluationCommercial
ExecuTorch
ExecuTorch

On-device AI across mobile, embedded and edge for PyTorch

Machine Learning Model TrainingOpen Source
Test Infra

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

Bi DashboardsOpen Source

Company Profile & Strategy

PyTorch publishes 2 products at pytorch.org, including Captum and Bayesian Optimization Library (botorch). Captum: Captum is a library for model interpretability in PyTorch.

Open-Source Footprint

9 repositories142.5K stars0 contributors

Save & manage the vendor's products in Productapp.

Save and manage this vendor's products in Productapp.

Start free trialWebsite