is there library for mesh analysis with jax support?

[jakubMitura14]

Hello I am developing some model, where part will be CNN part graph neural network and part based on mesh analysis - and now my question is is there some library that support mesh data structure sth like https://pytorch3d.org/ and will play well with jax (that I could run backpropagation fully through the whole model)

[adam-hartshorne]

https://github.com/ml-for-gp/jaxgptoolbox

[jakubMitura14]

thanks!