NVIDIA Corporation

All

540 repositories

cuEquivariance
Public
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.
Python
•8•176•4•6•Updated Jan 30, 2025Jan 30, 2025
cccl
Public
CUDA Core Compute Libraries
cpp hpc gpu modern-cpp parallel-computing cuda nvidia gpu-acceleration cuda-kernels gpu-computing
C++
•
Other
•183•1.4k•908•130•Updated Jan 30, 2025Jan 30, 2025
Fuser
Public
A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
C++
•
Other
•55•295•199•134•Updated Jan 30, 2025Jan 30, 2025
Megatron-LM
Public
Ongoing research training transformer models at scale
transformers model-para large-language-models
Python
•
Other
•2.5k•11k•173•155•Updated Jan 30, 2025Jan 30, 2025
Megatron-Energon
Public
Megatron's multi-modal data loader
Python
•
Other
•18•160•9•5•Updated Jan 30, 2025Jan 30, 2025
k8s-device-plugin
Public
NVIDIA device plugin for Kubernetes
kubernetes
Go
•
Apache License 2.0
•652•3k•136•35•Updated Jan 30, 2025Jan 30, 2025
numbast
Public
Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.
cuda numba
Python
•
Apache License 2.0
•8•32•16•4•Updated Jan 30, 2025Jan 30, 2025
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.7k•13k•49•97•Updated Jan 30, 2025Jan 30, 2025
spark-rapids-jni
Public
RAPIDS Accelerator JNI For Apache Spark
Cuda
•
Apache License 2.0
•68•47•72•16•Updated Jan 30, 2025Jan 30, 2025
nv-ingest
Public
NVIDIA Ingest is an early access set of microservices for parsing hundreds of thousands of complex, messy unstructured PDFs and other enterprise documents into metadata and text to embed into retrieval systems.
Python
•
Apache License 2.0
•201•2.4k•69•14•Updated Jan 30, 2025Jan 30, 2025
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
•206•594•313•43•Updated Jan 30, 2025Jan 30, 2025
VisRTX
Public
NVIDIA OptiX based implementation of ANARI
C++
•
Other
•30•249•5•0•Updated Jan 30, 2025Jan 30, 2025
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
•91•1.2k•36•3•Updated Jan 30, 2025Jan 30, 2025
cudaqx
Public
Accelerated libraries for quantum-classical computing built on CUDA-Q.
C++
•
Other
•14•19•8•5•Updated Jan 30, 2025Jan 30, 2025
warp
Public
A Python framework for high performance GPU simulation and graphics
Python
•
Other
•258•4.5k•107•4•Updated Jan 30, 2025Jan 30, 2025
bionemo-framework
Public
BioNeMo Framework: For building and adapting AI models in drug discovery at scale
machine-learning gpu pytorch drug-discovery
Jupyter Notebook
•
Other
•33•272•28•50•Updated Jan 30, 2025Jan 30, 2025
cloud-native-docs
Public
Documentation repository for NVIDIA Cloud Native Technologies
kubernetes containers kubernetes-operator
PowerShell
•
Apache License 2.0
•25•20•3•9•Updated Jan 30, 2025Jan 30, 2025
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
•191•1.4k•1•0•Updated Jan 30, 2025Jan 30, 2025
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
•352•2.1k•169•52•Updated Jan 30, 2025Jan 30, 2025
k8s-kata-manager
Public
Go
•
Apache License 2.0
•7•18•2•0•Updated Jan 30, 2025Jan 30, 2025
k8s-driver-manager
Public
The NVIDIA Driver Manager is a Kubernetes component which assist in seamless upgrades of NVIDIA Driver on each node of the cluster.
Shell
•
Apache License 2.0
•13•34•2•1•Updated Jan 30, 2025Jan 30, 2025
nvsdm
Public
NVSwitch Device Monitoring API.
Apache License 2.0
•0•2•0•0•Updated Jan 30, 2025Jan 30, 2025
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
•21•80•3•21•Updated Jan 30, 2025Jan 30, 2025
vgpu-device-manager
Public
NVIDIA vGPU Device Manager manages NVIDIA vGPU devices on top of Kubernetes
Go
•
Apache License 2.0
•15•124•0•2•Updated Jan 30, 2025Jan 30, 2025
mig-parted
Public
MIG Partition Editor for NVIDIA GPUs
Go
•
Apache License 2.0
•42•185•20•12•Updated Jan 30, 2025Jan 30, 2025
cuda-python
Public
CUDA Python: Performance meets Productivity
Python
•
Other
•91•1.1k•89•14•Updated Jan 30, 2025Jan 30, 2025
JAX-Toolbox
Public
JAX-Toolbox
Jupyter Notebook
•
Apache License 2.0
•56•279•80•39•Updated Jan 30, 2025Jan 30, 2025
NeMo-Run
Public
A tool to configure, launch and manage your machine learning experiments.
Python
•
Apache License 2.0
•33•108•23•12•Updated Jan 30, 2025Jan 30, 2025
NV-Kernels
Public
Ubuntu kernels which are optimized for NVIDIA server systems
C
•
Other
•17•32•0•15•Updated Jan 30, 2025Jan 30, 2025
NVFlare
Public
NVIDIA Federated Learning Application Runtime Environment
python decentralized pet privacy-protection federated-learning federated-analytics federated-computing
Python
•
Apache License 2.0
•182•681•12•11•Updated Jan 30, 2025Jan 30, 2025