edward-io Goto Github PK

builder

Continuous builder and binary build scripts for pytorch

captum

Model interpretability and understanding for PyTorch

clip

Contrastive Language-Image Pretraining

convnext

Code release for ConvNeXt model

kernl

Kernl lets you run Pytorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

low_cost_robot

metrics

Machine learning metrics for distributed, scalable PyTorch applications.

mingpt

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

multimodal

TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

recipes

Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research techniques without significant engineering overhead.Specifically, recipes aims to provide- Consistent access to pre-trained SOTA models ready for production- Reference implementations for SOTA research reproducibility, and infrastructure to guarantee correctness, efficiency, and interoperability.

rfcs

PyTorch RFCs (experimental)

rich

Rich is a Python library for rich text and beautiful formatting in the terminal.

stable-diffusion

tnt

A lightweight library for PyTorch training tools and utilities

torcheval1

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.

torchsnapshot-1

A light-weight library for adding fault tolerance to large-scale PyTorch distributed training workloads.

triton

Development repository for the Triton language and compiler

vision

Datasets, Transforms and Models specific to Computer Vision