-
/home $ java -jar org.eclipse.osgi_3.11.1.jar -console
Exception in thread "main" java.lang.ClassNotFoundException: Cannot load class:
org.eclipse.core.runtime.adaptor.Ecl…
-
I am under the impression that Elegy is not really an NN framework by itself. Instead, it wraps NN frameworks like Treex/Equinox/Flax/Haiku to automate the training loop and provide handy utilities li…
-
Hi everyone, thanks for the great library and apologies in advance for this basic question.
I'm trying to find the true minimum of a small neural network, and I thought of using a solver from `optimi…
-
I am creating masked convolution in equinox like below :
```
import equinox as eqx
import jax
import jax.numpy as jnp
from typing import Optional
from jaxtyping import Array, Float
class…
-
- [ ] Have it support classical treeverse, for improved results when using `equinox.internal.scan(..., kind="checkpointed")`.
- [ ] Have it support saving the residuals of each step, rather than just…
-
### Description
When trying to xmap an axis and to use a SerialLoop (in order to reduce memory usage) on a function that evaluates a neural network the following error is raised:
```
NameError: un…
-
[Equinox](https://docs.kidger.site/equinox/) is a minimalistic jax nn library. Currently, evosax's "parameter_reshaper" doesn't seem to support it, as the returned modules will have an population (bat…
-
This is really an exploratory question about implementing a Bayesian layer in Equinox. My research requires Bayesian neural networks, and I've been playing with Equinox to see how it stacks up to Flax…
-
### Описание ошибки
При описание параметра функции как ДокументОбъект или ДокументСсылка в автодополнение подставляются реквизиты из всех документов, ошибка при обращение к реквизиту уникальному для …
-
There seems to be a race to find the best implementation (currently 4) of KANs on the PyTorch side of the ML universe but there is a dearth of efforts on the JAX side (only 1 with flax). The flax impl…