drjit
Dr.Jit: A Just-In-Time Compiler for Differentiable Rendering
Decision gist · record as of 2026-08-14
Yes, if you are working on differentiable rendering, physics simulation, or large sparse computation graphs where ML frameworks struggle. The active maintenance, permissive license, and lack of known vulnerabilities support adoption. Medium install friction is acceptable given the specialized use case and broad platform coverage. Not recommended for general machine learning; use JAX, PyTorch, or TensorFlow instead.AI-flagged interpretation of the facts on this page — verify before relying
Before you install
- Requires Python >=3.8; GPU support (Metal on macOS, CUDA elsewhere) is optional but recommended for performance.
- Medium install friction due to compiled wheels for multiple Python versions and architectures (cp310–cp313, arm64/x86_64/Windows).
- Active maintenance with a release 7 days old and recent commits; no known vulnerabilities.
License · maintenance · safety
permissive license (permissive) — Permissive BSD license allows commercial and private use with minimal restrictions, making it suitable for research and production deployments.
last release 2026-08-07 (7 days) · last repo commit 2026-08-14 · 795 stars
0 known vulnerabilities (OSV.dev, 2026-08-14) · 207,914 downloads/mo, #9,539 on PyPI
Alternatives
Verify before relying
pip install drjit
import drjit as dr
# Create and trace a computation
x = dr.arange(dr.Float, 10)
y = x + 1- Specific performance characteristics compared to JAX, TensorFlow, or PyTorch for large sparse computation graphs.
- Whether the package is actively maintained beyond the Mitsuba 3 renderer project.
- Availability and quality of documentation beyond readthedocs.io.
- Practical limits on computation graph size and complexity.
What it is and what it does
Dr.Jit is a specialized just-in-time compiler designed to handle large, sparse computation graphs typical of differentiable rendering and similar embarrassingly parallel workloads. Unlike general machine learning frameworks, it records arithmetic operations into a computation graph, then JIT-compiles that graph into fused kernels targeting GPUs (Metal on macOS, CUDA elsewhere) or CPUs with vector instruction sets (AVX512, NEON) via LLVM. It optionally computes derivatives using forward or reverse-mode automatic differentiation, with both tracing and differentiation producing specialized code.
The package originated as the numerical foundation of Mitsuba 3, a differentiable Monte Carlo renderer, but is general-purpose and can be used without JIT compilation as a header-only vector library. It supports both C++17 and Python, allowing code to be developed in either language or both simultaneously, with joint tracing and differentiation across language boundaries. Dr.Jit includes a mathematical library with transcendental functions and types like vectors, matrices, complex numbers, and quaternions, and handles custom data structures, side effects, and polymorphism.
Use it for
- Differentiable rendering pipelines where computation graphs contain millions of elementary operations that would overwhelm ML framework compilers.
- Physics simulations requiring automatic differentiation and GPU acceleration without the overhead of general-purpose ML frameworks.
- Embarrassingly parallel numerical computations that benefit from vectorization and JIT compilation to specialized kernels.
- Research combining Python prototyping with C++ performance-critical code, jointly traced and differentiated.
- Monte Carlo methods and inverse problems where reverse-mode AD is essential for gradient computation.
Worth the install?
AI-flagged interpretation of the facts on this page. Verify before relying on it.
Yes, if you are working on differentiable rendering, physics simulation, or large sparse computation graphs where ML frameworks struggle.
The active maintenance, permissive license, and lack of known vulnerabilities support adoption. Medium install friction is acceptable given the specialized use case and broad platform coverage. Not recommended for general machine learning; use JAX, PyTorch, or TensorFlow instead.
Install
drjit on PyPI
Before you install
Medium install friction due to compiled wheels for multiple Python versions and architectures (cp310–cp313, arm64/x86_64/Windows). Active maintenance with a release 7 days old and recent commits; no known vulnerabilities.
Requires Python >=3.8; GPU support (Metal on macOS, CUDA elsewhere) is optional but recommended for performance.
License in practice
Permissive BSD license allows commercial and private use with minimal restrictions, making it suitable for research and production deployments.
Quickstart
pip install drjit
import drjit as dr
# Create and trace a computation
x = dr.arange(dr.Float, 10)
y = x + 1Verify before relying
- Specific performance characteristics compared to JAX, TensorFlow, or PyTorch for large sparse computation graphs.
- Whether the package is actively maintained beyond the Mitsuba 3 renderer project.
- Availability and quality of documentation beyond readthedocs.io.
- Practical limits on computation graph size and complexity.
Package facts
| License | permissive license permissive |
| Python support | Supports the current Python release >=3.8 |
| Install friction | Medium. Platform-specific wheel |
| Runtime dependencies | 1 packagetyping_extensions |
| Maintenance | Actively maintained 7 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 207,914 / month, #9,539 on PyPI 30-day window, as of 2026-08-14 |
| Known vulnerabilities | None known OSV.dev, checked 2026-08-14 |
| Classifiers | License :: OSI Approved :: BSD License |
Evidence: drjit-1.5.0-cp310-cp310-macosx_11_0_arm64.whl; drjit-1.5.0-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl; drjit-1.5.0-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl; drjit-1.5.0-cp310-cp310-win_amd64.whl; drjit-1.5.0-cp311-cp311-macosx_11_0_arm64.whl; drjit-1.5.0-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.whl; drjit-1.5.0-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl; drjit-1.5.0-cp311-cp311-win_amd64.whl; drjit-1.5.0-cp312-abi3-macosx_11_0_arm64.whl; drjit-1.5.0-cp312-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl; drjit-1.5.0-cp312-abi3-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl; drjit-1.5.0-cp312-abi3-win_amd64.whl; drjit-1.5.0-cp312-cp312-macosx_11_0_arm64.whl; drjit-1.5.0-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.whl; drjit-1.5.0-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl; drjit-1.5.0-cp312-cp312-win_amd64.whl; drjit-1.5.0-cp313-cp313-macosx_11_0_arm64.whl; drjit-1.5.0-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.whl; drjit-1.5.0-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl; drjit-1.5.0-cp313-cp313-win_amd64.whl
Tags
Let your AI agent find packages like this
Example. Real query, live index.
You found this page by searching. An agent finds it by wishing: SkillFed indexes 14,416 PyPI packages by what they can do, searchable in plain language.
wish › “jit compiler differentiable rendering”
- drjitDr.Jit is a just-in-time compiler for differentiable and ordinary…
- mitsubaMitsuba 3 is a research-oriented rendering system that simulates…
- jaxJAX is a Python library for automatic differentiation, XLA…
Give your agent the search over MCP, or paste the wish link into any chat.
More Scientific/Engineering packages
NumPy provides an N-dimensional array object and a comprehensive suite of mathematical, linear algebra, Fourier transform, and random number functions for scientific computing in Python.
pandas provides fast, flexible data structures (Series and DataFrame) for loading, cleaning, transforming, and analyzing labeled or relational data in Python.
scipy provides numerical algorithms for mathematics, science, and engineering—including optimization, integration, linear algebra, Fourier transforms, signal and image processing, and ODE solvers—built on numpy arrays.
scikit-learn provides a comprehensive Python library for supervised and unsupervised machine learning, including classification, regression, clustering, dimensionality reduction, and model evaluation tools built on NumPy and SciPy.
Install it if you need to train, evaluate, or deploy supervised or unsupervised learning models.
dill extends Python's pickle module to serialize and deserialize a much wider range of Python objects, including functions, lambdas, classes, and interpreter sessions, to byte streams for storage or network transmission.
Multiprocess is an enhanced fork of Python's standard multiprocessing library that uses dill for better serialization, allowing you to spawn processes with a threading-like API and share complex objects between them.
Install it if you use multiprocessing and encounter pickle serialization limits with lambdas or complex objects.
See also numba · nvidia-nvjitlink-cu12 · nvidia-nvjitlink · jax · jax-cuda12-pjrt · jax-cuda13-pjrt · nvidia-cuda-nvrtc-cu11 · numdifftools · torch-c-dlpack-ext · pyston