Fun exercise… but for such use cases, people use Casadi and DrJit

https://github.com/casadi/casadi https://github.com/mitsuba-renderer/drjit

DrJit is made by same author of pybind11 and nanobind.