daxiongshu

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

stylegan

StyleGAN - Official TensorFlow Implementation

arrow

Apache Arrow is a multi-language toolbox for accelerated data interchange and in-memory processing

stylegan2

StyleGAN2 - Official TensorFlow Implementation

catboost

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

TensorRT-LLM

TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.

tiny-gpu

A minimal GPU design in Verilog to learn how GPUs work from the ground up

FasterTransformer

Transformer related optimization, including BERT, GPT

ChatRTX

A developer reference project for creating Retrieval Augmented Generation (RAG) chatbots on Windows using TensorRT-LLM

scanpy

Single-cell analysis in Python. Scales to >1M cells.

lm-format-enforcer

Enforce the output format (JSON Schema, Regex etc) of a language model

NVTabular

NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.

cuda-python

CUDA Python Low-level Bindings

raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

tensorrtllm_backend

The Triton TensorRT-LLM Backend

NeMo-Curator

Scalable toolkit for data curation

GLUE

Graph-linked unified embedding for single-cell multi-omics data integration

Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

TuRBO

bbo_challenge_starter_kit

Starter kit for the black box optimization challenge at Neurips 2020

rapids-ai-BBO-2nd-place-solution

neurips2021_multimodal_topmethods

genedisco

GeneDisco is a benchmark suite for evaluating active learning algorithms for experimental design in drug discovery.

neurips2021_multimodal_viash

Diversity-Sampling

GPU-accelerated algorithm for subsampling datasets while preserving diversity

cython-bind-c-example-2020

It shows you how to create a package with cython and c/cpp.

CoAtGIN

CoAtGIN: Marrying Convolution and Attention for Graph-based Molecule Property Prediction

ALAE

[CVPR2020] Adversarial Latent Autoencoders