Luca Grementieri

Add support for float inputs in trimesh.grouping.boolean_rows

1

The function [trimesh.grouping.boolean_rows](https://trimesh.org/trimesh.grouping.html#trimesh.grouping.boolean_rows) is described as a very general function to find matching rows in arbitrary matrices but internally it converts the arrays to the type `np.int64`. Therefore, if the...

Install PyNomad in virtual environment

3

The current installation instructions make very hard to install PyNomad in virtual enviroment, in fact the `--user` option in the line `COMMAND python setup_PyNomad.py ${CMAKE_BINARY_DIR} ${NOMAD_VERSION} install` inside `interfaces/PyNomad/CMakeLists.txt` overwrite...

Wrong values in tensors for JAX function executed in C++ as xla::HloModule

4

I am following the guide published at https://github.com/google/jax/issues/5337 to run a JAX program in C++. The published example (https://github.com/google/jax/tree/main/examples/jax_cpp) works as expected, but if in the exported function I define...

Add support for jaxtyping

6

This PR adds support for jaxtyping annotations preserving all the features and checks on tensor dimensions. The PR doesn't update the README, since it could become messy very easily. I'll...

Use jaxtyping to enrich type annotations

5

I just discovered the [jaxtyping](https://github.com/google/jaxtyping) library and I think it could be an interesting alternative to the current typing system proposed by jax_dataclasses. jaxtyping supports variable-size axes and symbolic expressions...