Yarden As

Results 1 issues of Yarden As

The following code seems to fail ``` import numpy as np from torch import nn import torch import jax import equinox as eqx print(torch.__version__, eqx.__version__) random_input = jax.random.normal(jax.random.PRNGKey(1), (10, 90,...