Peter Fackeldey

Hi @henryiii @HDembinski , I assume the following is related, if not please correct me and I open a fresh new issue... We noticed in the scope of our analysis...

Thank you very much for your reply! Just to make sure I understand you correctly: I would like to serialise the following `eqx.Module`, and load it again without the need...

Thank you very much! Alright, that makes sense to me :) Just as a side note: I found out that `tf.keras.Model.save` serializes all functions/classes/... to some well-defined dict-structures, which can...

Hi @HDembinski, thank you very much for your reply :) Indeed it is possible to just use first order minimizer and to compute the hessian at the minimum with JAX...

> I think the proper thing is to support directly functions which accept a pytree instead of an array, as a generalization of the array-call mode that iminuit supports. Then...

I've managed to rephrase my compute task from a `array of PyTrees` to `PyTree of arrays` paradigm. Then I could use `jax.lax.scan` similar to the example you provided on the...

Thank you very much for your reply! I'm looking already into `eqx.tree_equal` but I am not yet fully understanding how to use it in a JIT'd function: Essentially I want...

**Update:** I think I just found the solution: ```python def f(t1, t2): return eqx.error_if(t1, eqx.tree_equal(t1, t2) != True, "Error, t1 and t2 are unequal") f(t1, t2) # -> Error f(t1,...

Great! If that's fine for you I will close the issue as I consider all my questions as solved :)

Hi @JuanPedroGHM, I'd like to measure the energy consumption of a dask cluster. Now keep in mind, that these are typically distributed, so I want to track the energy consumption...