-
Hello,
I'm a new user of the equinox library so maybe my problem is obvious to solve but I cannot figure out how to do that properly.
Basically I'm trying to pass a vmapped submodule as a field …
-
I've been trying to use `OptaxSolver` to perform a simple function minimization, since I want to differentiate through it's solution (the fixed point of the solver), but ran into an issue I'm not fami…
-
I run all the command at 1st time successfully.
When I run below command for 2nd time:
```
python -m alphageometry \
--alsologtostderr \
--problems_file=$(pwd)/examples.txt \
--problem_name=orth…
-
Hi,
Do you think the code may run with new version of brax and jax ?
I have:
brax 0.10.5
brax-jumpy 0.1.0
flax 0.8.4
jax …
-
Hello. I tried experimenting with the rules so I could train yachzotron for a slightly different game. Unfortunately, call
`python3 yahtzotron/cli.py train -o custom.pkl --ruleset yatzy_modified`
fa…
-
I've run into a strange behavior, and I'm unsure if its a bug or if I'm doing something wrong, and I wasn't able to find any clarification in the docs: it seems that the weights of my network get init…
-
The [Adafactor paper](https://arxiv.org/pdf/1804.04235.pdf) (section 6) suggests "update clipping" instead of the usual "gradient clipping".
![image](https://user-images.githubusercontent.com/9633…
-
See the last few code boxes of https://colab.research.google.com/drive/1U_qMlcQfD1Dxe-_V9cHpNOA2iF8X1jYg#scrollTo=emIvHzxwQtzj&line=21&uniqifier=1
**Adam#Lion optimizer coding**
```python
impor…
-
Example:
```python
from functools import partial
import jax
import jax.numpy as jnp
import optax
@partial(jax.jit, static_argnames=("optimizer",))
def train_step(optimizer, opt_state):
…
-
[University of Virginia - Deep Learning (Jax+Flax) ](https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/JAX/tutorial2/Introduction_to_JAX.html)
> [Available Package](https://gith…