NVIDIA Corporation
Change the repository type filter
All
Repositories list
655 repositories
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffDock, MACE, Allegro and NEQUIP, based on equivariant neural networks. Also includes kernels for accelerated structure prediction.
- CUDA Core Compute Libraries
- C++ and Python support for the CUDA Quantum programming model for heterogeneous quantum-classical workflows
- Ongoing research training transformer models at scale
- TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.
- NVIDIA device plugin for Kubernetes
- BioNeMo Framework: For building and adapting AI models in drug discovery at scale
- GPU accelerated decision optimization
- A unified library of SOTA model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.
- Examples for Recommenders - easy to train and deploy on accelerated infrastructure.
- AIStore: scalable storage for AI applications