David Cottrell

yes I use it no problem on 3.11

I was curious about this, minor clean up and added another test function ```python # https://github.com/patrick-kidger/equinox/issues/576 import equinox as eqx import jax import jax.numpy as jnp from jax import grad,...

@ToshiyukiBandai slower gpu perf is typical for small test problems so that is fine. The relative perf of the equinox module is interesting and hopefully we can figure it out.

@smorad fwiw, I did this kind of thing in some util I made once. ```python import jax import numpy as np def get_summary_info(model): """An alternative repr useful for initial debugging"""...

I think you are misunderstanding. This is about the inputs to init which, if they were to be interpreted as code, would make for and incredibly inconvenient paradigm in the...

Wait ... isn't this (in examples) the pattern I'm asking about? I somehow didn't see that. Ok this makes more sense. https://docs.kidger.site/equinox/examples/serialisation/ I'll post some interesting hack in a sec...

This is probably a horror show, and I'm not super familiar with clever ways to do things in pytrees, but "it works" (maybe) as an example I think. The only...

Thanks, ok I'll probably kick it along and try to make it more sane with your suggestions. The dict/module to special dict is nice. I guess like you said with...

Incoporated all the suggestions I think except for functions serialization. Haven't change the names or anything yet. Do you have a list of test modules somewhere? I mean a set...

@patrick-kidger Have added a bunch of tests and some hacky serialization thing because I need that. But this is probably orthogonal to this stuff. There are a number of ugly...