NVIDIA/warp
Oct 2025 – Oct 2025
1 Month active
Languages Used
C++Python
Technical Skills
Automatic DifferentiationCUDAFFI (Foreign Function Interface)JAXKernel Development
Mehdi Ataei
PROFILE
Mehdi Ataei
Developed and integrated automatic differentiation support for Warp kernels within the NVIDIA/warp repository, enabling seamless use of differentiable kernels in JAX-based machine learning and scientific computing workflows. The work involved implementing adjoint computation in the jax_kernel component, allowing gradient-based optimization directly on Warp kernels. Leveraging expertise in C++, Python, CUDA, and FFI, the developer ensured robust integration by providing comprehensive documentation and unit tests to validate the new feature. This addition reduces friction for users seeking to combine Warp’s high-performance kernels with JAX’s differentiation capabilities, laying a foundation for more flexible and efficient experimentation in computational pipelines.
Overall Statistics
Feature vs Bugs
100%Features
Repository Contributions
1Total
Bugs
0
Commits
1
Features
1
Lines of code
924
Activity Months1
Your Network
128 peopleWork History
October 2025
1 Commits • 1 Features
Oct 1, 2025October 2025 Monthly Summary for NVIDIA/warp: Delivered JAX Warp Kernel Automatic Differentiation feature with adjoint computation, plus docs and tests. This enables differentiable Warp kernels within JAX for ML and scientific computing workflows, reducing integration friction and accelerating experimentation.
1 Commits • 1 Features
Oct 1, 2025October 2025 Monthly Summary for NVIDIA/warp: Delivered JAX Warp Kernel Automatic Differentiation feature with adjoint computation, plus docs and tests. This enables differentiable Warp kernels within JAX for ML and scientific computing workflows, reducing integration friction and accelerating experimentation.
October 2025
Activity
Loading activity data...
Quality Metrics
Correctness100.0%
Maintainability100.0%
Architecture100.0%
Performance80.0%
AI Usage20.0%
Skills & Technologies
Programming Languages
C++Python
Technical Skills
Automatic DifferentiationCUDAFFI (Foreign Function Interface)JAXKernel Development
Repositories Contributed To
Overview of all repositories you've contributed to across your timeline