NVIDIA Corporation

All

518 repositories

JAX-Toolbox
Public
JAX-Toolbox
Jupyter Notebook
•
Apache License 2.0
•45•241•114•39•Updated Oct 31, 2024Oct 31, 2024
gpu-operator
Public
NVIDIA GPU Operator creates, configures, and manages GPUs in Kubernetes
kubernetes gpu cuda nvidia
Go
•
Apache License 2.0
•295•1.8k•273•24•Updated Oct 31, 2024Oct 31, 2024
cccl
Public
CUDA Core Compute Libraries
cpp hpc gpu modern-cpp parallel-computing cuda nvidia gpu-acceleration cuda-kernels gpu-computing
C++
•
Other
•157•1.2k•799•52•Updated Oct 31, 2024Oct 31, 2024
doca-sosreport
Public
A unified tool for collecting system logs and other debug information
Python
•
GNU General Public License v2.0
•542•1•0•2•Updated Oct 31, 2024Oct 31, 2024
NeMo
Public
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
machine-translation tts speech-synthesis neural-networks deeplearning speaker-recognition asr multimodal speech-translation large-language-models
Python
•
Apache License 2.0
•2.5k•12k•49•93•Updated Oct 31, 2024Oct 31, 2024
gpu-driver-container
Public
The NVIDIA GPU driver container allows the provisioning of the NVIDIA driver through the use of containers.
Shell
•
Apache License 2.0
•33•67•9•14•Updated Oct 31, 2024Oct 31, 2024
Megatron-LM
Public
Ongoing research training transformer models at scale
transformers model-para large-language-models
Python
•
Other
•2.3k•10k•123•147•Updated Oct 31, 2024Oct 31, 2024
nvidia-container-toolkit
Public
Build and run containers leveraging NVIDIA GPUs
Go
•
Apache License 2.0
•259•2.4k•302•25•Updated Oct 31, 2024Oct 31, 2024
topograph
Public
A toolkit for generation of cluster network topology.
Go
•
Apache License 2.0
•0•3•0•3•Updated Oct 31, 2024Oct 31, 2024
warp
Public
A Python framework for high performance GPU simulation and graphics
Python
•
Other
•239•4.2k•49•3•Updated Oct 31, 2024Oct 31, 2024
numba-cuda
Public
Python
•
BSD 2-Clause "Simplified" License
•7•35•14•7•Updated Oct 31, 2024Oct 31, 2024
spark-rapids-jni
Public
RAPIDS Accelerator JNI For Apache Spark
Cuda
•
Apache License 2.0
•65•36•79•9•Updated Oct 31, 2024Oct 31, 2024
cuda-quantum
Public
C++ and Python support for the CUDA Quantum programming model for heterogeneous quantum-classical workflows
python cpp quantum quantum-computing hacktoberfest quantum-programming-language quantum-algorithms quantum-machine-learning unitaryhack
C++
•
Other
•182•508•274•36•Updated Oct 31, 2024Oct 31, 2024
Fuser
Public
A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
C++
•
Other
•52•263•248•137•Updated Oct 31, 2024Oct 31, 2024
holodeck
Public
Holodeck is a project to create test environments optimised for GPU projects.
Go
•
Apache License 2.0
•5•9•3•1•Updated Oct 31, 2024Oct 31, 2024
knavigator
Public
knavigator is a development, testing, and optimization toolkit for AI/ML scheduling systems at scale on Kubernetes.
Go
•
Apache License 2.0
•9•51•0•2•Updated Oct 31, 2024Oct 31, 2024
spark-rapids
Public
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
big-data gpu rapids spark
Scala
•
Apache License 2.0
•234•804•1.4k•25•Updated Oct 31, 2024Oct 31, 2024
cuCollections
Public
datastructures cpp gpu cuda hashmap cpp17 hashset hashtable
C++
•
Apache License 2.0
•88•480•65•10•Updated Oct 31, 2024Oct 31, 2024
TransformerEngine
Public
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
Python
•
Apache License 2.0
•321•1.9k•139•46•Updated Oct 31, 2024Oct 31, 2024
NeMo-Curator
Public
Scalable data pre processing and curation toolkit for LLMs
python data data-processing data-preparation deduplication data-quality data-curation data-prep fine-tuning fast-data-processing
Jupyter Notebook
•
Apache License 2.0
•76•560•54•25•Updated Oct 31, 2024Oct 31, 2024
bionemo-framework
Public
BioNeMo Framework: For building and adapting AI models in drug discovery at scale
machine-learning gpu pytorch drug-discovery
Python
•
Other
•2•30•19•43•Updated Oct 31, 2024Oct 31, 2024
NeMo-Run
Public
A tool to configure, launch and manage your machine learning experiments.
Python
•
Apache License 2.0
•10•56•15•5•Updated Oct 31, 2024Oct 31, 2024
aistore
Public
AIStore: scalable storage for AI applications
kubernetes sds erasure-coding object-storage software-defined multiple-backends batch-jobs distributed-shuffle linear-scalability etl-offload
Go
•
MIT License
•171•1.3k•1•0•Updated Oct 31, 2024Oct 31, 2024
TensorRT-Incubator
Public
Experimental projects related to TensorRT
MLIR
•12•75•47•21•Updated Oct 30, 2024Oct 30, 2024
DALI
Public
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
python machine-learning deep-learning neural-network mxnet gpu image-processing pytorch gpu-tensorflow data-processing
C++
•
Apache License 2.0
•619•5.1k•201•36•Updated Oct 30, 2024Oct 30, 2024
NeMo-Aligner
Public
Scalable toolkit for efficient model alignment
Python
•
Apache License 2.0
•71•594•61•39•Updated Oct 30, 2024Oct 30, 2024
k8s-nim-operator
Public
An Operator for deployment and maintenance of NVIDIA NIMs and NeMo microservices in a Kubernetes environment.
Go
•
Apache License 2.0
•9•50•1•15•Updated Oct 30, 2024Oct 30, 2024
cuda-python
Public
CUDA Python Low-level Bindings
Python
•
Other
•72•875•78•5•Updated Oct 30, 2024Oct 30, 2024
cuDecomp
Public
An Adaptive Pencil Decomposition Library for NVIDIA GPUs
fft pencil-decomposition
C++
•
Other
•8•57•1•0•Updated Oct 30, 2024Oct 30, 2024
MatX
Public
An efficient C++17 GPU numerical computing library with Python-like syntax
hpc gpu cuda gpgpu gpu-computing
C++
•
BSD 3-Clause "New" or "Revised" License
•82•1.2k•34•4•Updated Oct 30, 2024Oct 30, 2024