Type Caster for linear algebra library Armadillo (CPU) and Bandicoot (GPU)

[r1cheu]

Armadillo and Bandicoot offer nearly identical syntax for linear algebra operations, making it easy to transition between CPU and GPU implementations.

It would be great if nanobind could support conversions between Python libraries like NumPy, JAX, etc., and these libraries directly.

[wjakob]

nanobind offers the ndarray interface that allows it to accept and return general nd-arrays from CPU/GPU frameworks. This could be used to implement bindings for libraries like armadillo or bandicoot. I suggest that you review the documentation of the ndarray feature.

[r1cheu]

thanks for quick reply. I’ll dig into the ndarray interface. I’d willing to implement a type caster, like the one for Eigen(though I’m quite new to template metaprogramming, and it may take quite a long time). Any advice?