NVIDIA / warp
A Python framework for accelerated simulation, data generation and spatial computing.
☆5,073Updated this week
Alternatives and similar repositories for warp
Users that are interested in warp are comparing it to the libraries listed below
Sorting:
- CUDA Python: Performance meets Productivity☆2,660Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,169Updated this week
- Multi-Joint dynamics with Contact. A general purpose physics simulator.☆9,564Updated this week
- Lightning fast C++/CUDA neural network framework☆4,014Updated 2 weeks ago
- Next-gen fast plotting library running on WGPU using the pygfx rendering engine☆1,271Updated last week
- A curated list of awesome Taichi applications, courses, demos and features.☆595Updated 11 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,551Updated this week
- NumPy & SciPy for GPU☆10,208Updated 2 weeks ago
- Tile primitives for speedy kernels☆2,352Updated this week
- Visualize streams of multimodal data. Free, fast, easy to use, and simple to integrate. Built in Rust.☆8,420Updated this week
- Development repository for the Triton language and compiler☆15,568Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,779Updated 3 weeks ago
- Tiny Differentiable Simulator is a header-only C++ and CUDA physics library for reinforcement learning and robotics with zero dependencie…☆1,287Updated 7 months ago
- NVIDIA PhysX SDK☆3,923Updated last month
- a programming library with geometric algorithms☆2,094Updated this week
- CUDA Templates for Linear Algebra Subroutines☆7,540Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,633Updated last month
- Gaia Physics Engine☆639Updated 6 months ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,132Updated this week
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,342Updated this week
- CUDA integration for Python, plus shiny features☆1,936Updated last week
- CUDA Core Compute Libraries☆1,636Updated this week
- TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and support state-of-the-art optimizati…☆10,508Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,041Updated this week
- Run PyTorch LLMs locally on servers, desktop and mobile☆3,580Updated this week
- Schedule-Free Optimization in PyTorch☆2,161Updated last month
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆32,246Updated this week
- A reactive notebook for Python — run reproducible experiments, query with SQL, execute as a script, deploy as an app, and version with gi…☆13,264Updated this week
- A modern model graph visualizer and debugger☆1,179Updated this week
- CUDA Library Samples☆1,934Updated this week