jamesr66a

codegen2-1B

stable-fast

An ultra lightweight inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

aws-neuron-sdk

Powering Amazon custom machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services

datasets

🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

dist_loopnest

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

fx_traceable_mypy

hub

Submission to https://pytorch.org/hub/

imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

invisible-watermark-gpu

python library for invisible image watermark (blind image watermark)

jamesr66a.github.io

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

julia

The Julia Programming Language

jupyter_core

Core Jupyter functionality

Megatron-LM

Ongoing research training transformer language models at scale, including: BERT & GPT-2

minGPT

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

mypy

Optional static typing for Python 3 and 2 (PEP 484)

ossci-job-dsl

Jenkins job definitions for OSSCI

PiPPy

Pipeline Parallelism for PyTorch

pytorch-dockerfiles

pytorch-struct

Fast, general, and tested differentiable structured prediction in PyTorch

rfcs

PyTorch RFCs (experimental)

torch2trt

An easy to use PyTorch to TensorRT converter

torchdistx

Torch Distributed Experimental

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

triton

Development repository for the Triton language and compiler

tutorials