feat: make root_vjp public
#157
+165
−4
Open
root_vjp public
#157