Các thư viện được viết bằng Cuda

instant-ngp

Nguyên thủy đồ họa thần kinh tức thì: NeRF nhanh như chớp và hơn thế nữa.
  • 13.4k
  • GNU General Public License v3.0

deep-high-resolution-net.pytorch

Dự án là một triển khai chính thức của bài viết CVPR2019 của chúng tôi "Học sâu về biểu diễn độ phân giải cao để ước tính tư thế con người".
  • 4.1k
  • MIT

cub

Nguyên thủy hợp tác cho CUDA C++. (của NVIDIA).
  • 1.5k
  • BSD 3-clause "New" or "Revised"

cugraph

cuGraph - Thư viện phân tích đồ thị RAPIDS.
  • 1.3k
  • Apache License 2.0

blocksparse

Nhân GPU hiệu quả cho phép nhân và tích chập ma trận thưa thớt khối.
  • 948
  • MIT

k2

Thuật toán FSA/FST, khả vi, với khả năng tương thích PyTorch..
  • 927
  • Apache License 2.0

nv-wavenet

Triển khai tham chiếu của suy luận wavenet tự hồi quy thời gian thực.
  • 700
  • BSD 3-clause "New" or "Revised"

nvParse

Trình phân tích cú pháp CSV nhanh, dựa trên gpu.
  • 550
  • Apache License 2.0

instant-ngp-Windows

Nguyên thủy đồ họa thần kinh tức thì: NeRF nhanh như chớp và hơn thế nữa.
  • 458
  • GNU General Public License v3.0

nccl-tests

Các bài kiểm tra NCCL.
  • 436
  • BSD 3-clause "New" or "Revised"

MegBA

MegBA: Thư viện phân tán dựa trên GPU để điều chỉnh gói quy mô lớn.
  • 368
  • Apache License 2.0

raft

RAFT chứa các thuật toán và nguyên hàm cơ bản được sử dụng rộng rãi cho khoa học dữ liệu, đồ thị và học máy. (bởi Rapidsai).
  • 278
  • Apache License 2.0

dietgpu

Triển khai GPU của bộ mã hóa và giải mã entropy ANS (hệ thống số bất đối xứng) tổng quát hóa nhanh, với các phần mở rộng để nén không mất dữ liệu các loại dữ liệu số và dữ liệu khác trong các ứng dụng HPC/ML..
  • 247
  • MIT

TorchPQ

Tìm kiếm hàng xóm gần nhất với lượng tử hóa sản phẩm trên GPU trong pytorch và cuda.
  • 173
  • MIT

Lantern

  • 163
  • BSD 3-clause "New" or "Revised"

RWKV-CUDA

Phiên bản CUDA của mô hình ngôn ngữ RWKV (https://github.com/BlinkDL/RWKV-LM).
  • 113

cuhnsw

Triển khai CUDA của thuật toán Đồ thị thế giới nhỏ có thể điều hướng theo cấp bậc.
  • 88
  • Apache License 2.0

CUB

KHO SỞ HỮU NÀY ĐÃ CHUYỂN ĐẾN github.com/nvidia/cub, ĐƯỢC PHÉP TỰ ĐỘNG TẠI ĐÂY..
  • 71
  • BSD 3-clause "New" or "Revised"

array-language-comparisons

So sánh các ngôn ngữ & thư viện mảng: APL, J, BQN, Q, Julia, R, NumPy, Nial, Futhark, SaC & ArrayFire..
  • 60
  • MIT

mish-cuda

Chức năng Kích hoạt Mish cho PyTorch (bởi JunnYu).
  • 48
  • MIT

rocm-examples

  • 39
  • MIT

xgboost-node

Chạy mô hình XGBoost và đưa ra dự đoán trong Node.js.
  • 33
  • GNU General Public License v3.0

CUDA-Guide

CUDA Guide.
  • 29

kobra

Công cụ trò chơi theo định hướng nghiên cứu khám phá các kỹ thuật kết xuất hiện đại (bởi vedavamadathil).
  • 21

CryptoGPU

Phân tích giá tiền điện tử.
  • 16

LSQR-CUDA

Đây là một triển khai LSQR-CUDA do Lawrence Ayers viết dưới sự giám sát của Stefan Guthe thuộc viện GRIS tại Technische Universität Darmstadt. Thư viện LSQR là tác giả của Chris Paige và Michael Saunders..
  • 11
  • MIT

SBNN

Mạng thần kinh nhị phân số ít dựa trên Hoạt động bit GPU (xem bài báo SC-19 của chúng tôi).
  • 10
  • GNU General Public License v3.0

GCGT

Mã nguồn của bài báo: Traversal Graph Traversal dựa trên GPU.
  • 7
  • MIT

FirstCollisionTimestepRarefiedGasSimulator

Trình giả lập này tính toán tất cả các giao điểm có thể có trong một dấu thời gian rất nhỏ đối với mô hình hạt.
  • 2

DOKSparse

các tenxơ DOK thưa thớt trên GPU, pytorch.
  • 1
  • MIT