My Portfolio - Personal Website

Focus on prompting and generating

RelBench: Relational Deep Learning Benchmark

Robust Speech Recognition via Large-Scale Weak Supervision

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

Graph Neural Network Library for PyTorch

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

⚡ Building applications with LLMs through composability ⚡

LlamaIndex (GPT Index) is a data framework for your LLM applications

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

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

Inference code for CodeLlama models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

Stable Diffusion web UI

An open platform for training, serving, and evaluating large language model for tool learning.

Generative Models by Stability AI

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs

A library for federated learning (a distributed machine learning process) in an enterprise environment.

An Open Source Machine Learning Framework for Everyone

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

High-Resolution Image Synthesis with Latent Diffusion Models

The GitHub Archive Program & Arctic Code Vault

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

Repository for benchmarking graph neural networks