Owen L

I think the function has to be hashable, but the module isn't, a simple fix is you can wrap it in a lambda (like https://github.com/patrick-kidger/equinox/issues/709#issuecomment-2150846937)

you can just look at the source code and see what it does https://github.com/patrick-kidger/equinox/blob/main/equinox/nn/_linear.py#L59 https://github.com/patrick-kidger/equinox/blob/main/equinox/nn/_misc.py#L33

The most surprising thing about this to me at first glance is that a trivial 3 layer MLP is almost 100% slower with equinox than with flax. Even without any...

I isolated some of the MLP specific code here: https://github.com/patrick-kidger/equinox/issues/928

@atantos, @frankier Thanks for agreeing to review this submission! This is the review thread for the paper. All of our communications will happen here from now on. As you can...

Hi @frankier, let me know if you have any questions or concerns with the editorial bot or the review. When you start by running ``` @editorialbot generate my checklist ```...

Hi @atantos, just wanted to make sure you didn't have any hurdles or questions for me about the editorialbot or the review since I know we just entered the period...

> Hi [@lockwo](https://github.com/lockwo) and thank you for the reminder. I’ve been down with a high fever for a few days, but I’m already recovering and should be able to resume...