#

CUDA

CUDA® is a parallel computing platform and programming model developed by NVIDIA for general computing on graphical processing units (GPUs). With CUDA, developers are able to dramatically speed up computing applications by harnessing the power of GPUs.

Here are 637 public repositories matching this topic...

vllm-project / vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

amd cuda inference pytorch transformer llama gpt rocm model-serving mlops llm inferentia llmops llm-serving trainium

Updated May 12, 2024
Python

numba

numba / numba

NumPy aware dynamic Python compiler using LLVM

python compiler numpy llvm parallel cuda numba

Updated May 12, 2024
Python

cupy / cupy

NumPy & SciPy for GPU

python gpu numpy cuda cublas scipy tensor cudnn rocm cupy cusolver nccl curand cusparse nvrtc cutensor nvtx cusparselt

Updated May 12, 2024
Python

catboost / 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.

python data-science machine-learning data-mining tutorial r big-data gpu cuda kaggle gbdt gbm gpu-computing decision-trees gradient-boosting coreml catboost categorical-features

Updated May 12, 2024
Python

replicate / cog

Containers for machine learning

docker machine-learning ai deep-learning containers tensorflow cuda pytorch

Updated May 10, 2024
Python

chainer / chainer

A flexible framework of neural networks for deep learning

python machine-learning deep-learning neural-network chainer gpu numpy cuda neural-networks cudnn cupy

Updated Aug 28, 2023
Python

NVIDIAGameWorks / kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

cuda pytorch artificial-intelligence neural-networks camera-api rasterization 3d-deep-learning differentiable-rendering differentiable-lighting

Updated Mar 15, 2024
Python

nvitop

XuehaiPan / nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

console monitoring gpu cuda nvidia curses nvml top command-line-tool htop nvidia-smi monitoring-tool process-monitoring gpu-monitoring resource-monitor

Updated May 8, 2024
Python

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

python deep-learning gpu cuda jittor

Updated May 8, 2024
Python

pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

machine-learning deep-learning cuda pytorch nvidia jetson tensorrt libtorch

Updated May 12, 2024
Python

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

computer-vision deep-learning neural-network cuda high-dimensional-data pytorch convolutional-neural-networks auto-differentiation semantic-segmentation space-time spatio-temporal-analysis 3d-vision 3d-convolutional-network sparse-tensors sparse-convolution high-dimensional-inference trilateral-filter 4d-convolutional-neural-network minkowski-engine sparse-tensor-network

Updated Mar 5, 2024
Python

CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

cuda pytorch ema triplet-loss label-smoothing focal-loss amsoftmax dice-loss mish lovasz-softmax partial-fc

Updated Sep 29, 2022
Python

inducer / pycuda

CUDA integration for Python, plus shiny features

python gpu array cuda scientific-computing gpu-computing multidimensional-arrays pycuda

Updated May 8, 2024
Python

pytorch / torchrec

Pytorch domain library for recommendation systems

deep-learning gpu cuda pytorch recommendation-system sharding recommender-system

Updated May 12, 2024
Python

pykeen

pykeen / pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

python machine-learning deep-learning cuda torch link-prediction knowledge-base-completion knowledge-graph-embeddings knowledge-graphs pykeen

Updated May 4, 2024
Python

sniklaus / 3d-ken-burns

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

python deep-learning cuda pytorch cupy

Updated Mar 1, 2024
Python

NVIDIA / TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

python machine-learning deep-learning gpu cuda pytorch jax fp8

Updated May 11, 2024
Python

viseron

roflcoopter / viseron

Self-hosted, local only NVR and AI Computer Vision software. With features such as object detection, motion detection, face recognition and more, it gives you the power to keep an eye on your home, office or any other place you want to monitor.

Updated May 8, 2024
Python

AlexiaJM / Deep-learning-with-cats

Deep learning with cats (^._.^)

cat deep-learning picture cuda gan

Updated May 16, 2020
Python

openppl-public / ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

open-source caffe deep-learning neural-network cuda pytorch quantization onnx

Updated Mar 28, 2024
Python

Created by Nvidia

Released June 23, 2007

Followers: 197 followers
Website: developer.nvidia.com/cuda-zone
Wikipedia: Wikipedia

Related Topics

nvcc