Xavier Gonzalez

icon indicating a website link https://www.linkedin.com/in/xavier-gonzalez-517b5262/

icon company @lindermanlab icon location Stanford University icon location PhD student in statistics at Stanford in the @lindermanlab

Results 5 comments of Xavier Gonzalez

Keyboard only use

Is there a shortcut for adding a cell above or below a given cell?

Autoregressive flow on jax

Has there been any movement on this issue? When I run this very simple code in a google colab: ``` import jax import jax.numpy as jnp import jax.nn as jnn...

[TPU] TypeError with JAX 0.3.17 in Google Collab

I am getting this error currently, even after beginning my code in colab with ``` # get the latest JAX and jaxlib !pip install --upgrade -q jax jaxlib # Colab...

Different results for dopri5 depending on loop vs solve

That's really interesting @lockwo, I wonder why this is!

Bug in how eqx.Module interacts with jax.tree.map

Thank you so much for your very helpful and fast reply!