Update README.md

0xSooki · web-flow · commit 2a276c1a4d00 · 2025-05-28T17:48:44.000+02:00
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# JAX Custom Operations with C++ and CUDA
+# JAX Custom Operations with Pybind11, C++ and CUDA
 
 A demonstration project showing how to extend JAX with custom C++ and CUDA operations using the XLA FFI (Foreign Function Interface). This project implements a simple mathematical operation `foo(a, b) = sum(a * (b + 1))` with both CPU and GPU kernels, complete with gradient support.
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-# JAX Custom Operations with C++ and CUDA`
	`1`	`+# JAX Custom Operations with Pybind11, C++ and CUDA`
`2`	`2`
`3`	`3`	A demonstration project showing how to extend JAX with custom C++ and CUDA operations using the XLA FFI (Foreign Function Interface). This project implements a simple mathematical operation `foo(a, b) = sum(a * (b + 1))` with both CPU and GPU kernels, complete with gradient support.
`4`	`4`