Ask

[GeorgyChistov]

Hello!

I would like to say thank you for your library. It is more clear than initial UniRep.

Could you add argument "backend" to evotune() function? It will be very helpful, because now it is not impossible to use GPU when evotune is executed.

Also, it will be cool to add environment variables of jax
XLA_PYTHON_CLIENT_ALLOCATOR=platform
XLA_PYTHON_CLIENT_MEM_FRACTION=.XX
XLA_PYTHON_CLIENT_PREALLOCATE=false
inside your code and maybe make it arguments for functions... Maybe I am newbie in jax, but it seems to be very useful.

Anyway, thank you for your work!