NVIDIA / warp
A Python framework for high performance GPU simulation and graphics
☆4,270Updated this week
Related projects ⓘ
Alternatives and complementary repositories for warp
- Lightning fast C++/CUDA neural network framework☆3,768Updated 2 months ago
- A lightweight library for portable low-level GPU computation using WebGPU.☆3,755Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,680Updated this week
- Multi-Joint dynamics with Contact. A general purpose physics simulator.☆8,247Updated this week
- A unified framework for 3D content generation.☆6,330Updated last month
- CoTracker is a model for tracking any point (pixel) on a video.☆3,850Updated last month
- NumPy & SciPy for GPU☆9,498Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆6,146Updated this week
- Mitsuba 3: A Retargetable Forward and Inverse Renderer☆2,092Updated this week
- Tile primitives for speedy kernels☆1,661Updated this week
- High-resolution models for human tasks.☆4,523Updated this week
- Implementation of Nougat Neural Optical Understanding for Academic Documents☆8,997Updated 7 months ago
- TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain…☆8,695Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,562Updated 3 weeks ago
- Gaia Physics Engine☆604Updated 2 weeks ago
- Solve puzzles. Learn CUDA.☆9,956Updated 2 months ago
- A reactive notebook for Python — run reproducible experiments, execute as a script, deploy as an app, and version with git.☆7,872Updated this week
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,678Updated this week
- A curated list of awesome Taichi applications, courses, demos and features.☆500Updated 5 months ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,712Updated this week
- Solve puzzles. Improve your pytorch.☆3,286Updated 4 months ago
- Schedule-Free Optimization in PyTorch☆1,900Updated 2 weeks ago
- A minimal GPU design in Verilog to learn how GPUs work from the ground up☆7,092Updated 3 months ago
- Official Repository for "Eureka: Human-Level Reward Design via Coding Large Language Models" (ICLR 2024)☆2,841Updated 6 months ago
- cuDF - GPU DataFrame Library☆8,461Updated this week
- lightweight, standalone C++ inference engine for Google's Gemma models.☆5,992Updated this week
- A collaboration friendly studio for NeRFs☆9,559Updated this week
- MLX: An array framework for Apple silicon☆17,358Updated this week
- A library for differentiable nonlinear optimization☆1,777Updated last week
- Dr.Jit — A Just-In-Time-Compiler for Differentiable Rendering☆594Updated this week