Xavier Gonzalez

Struggles with Gaussian Mixture Models in jax port of tensorflow_distributions

2

I am getting the strangest bugs when trying to make a Guassian mixture model class in the jax substrate of tfd, has anyone experienced this before or know what the...

Thanks so much for making a wonderful package! I am trying to download your code into Google Colabs. When I type `!pip install git+https://github.com/amirgholami/HessianFlow.git` I get the following error message:...

Different results for dopri5 depending on loop vs solve

4

I am trying to solve the Lorenz butterfly using diffrax, and in particular the `Dopri5` solver. I take two approaches. In of them, I simply solve over the entire time...

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

2

There seems to be a bug in how eqx.Module interacts with jax.tree.map Here is the repro: ``` import jax import equinox as eqx import jax.numpy as jnp from flax import...