Brian Patton
Brian Patton
I don't think we're currently investing many cycles in numerical integration, so if you want to contribute something, it won't be duplicated work. We are looking at "memory efficient MCMC"...
For now, you could call it within a @tf.function-decorated callable. Brian Patton | Software Engineer | [email protected] On Thu, Oct 10, 2019 at 10:37 PM Shashank Gupta wrote: > I...
OK I got things working with the jax nightly install (added the pip command to the ipynb). Probably should clear the outputs of the ipynb before merging to avoid a...
btw, a lot of credit to @midfield for help building this out.
@csuter fyi
Potentially we could only require dm-tree for non-JAX variants (TF and NumPy). Not sure if that solves your concern, though. On Sat, Oct 5, 2024 at 8:55 AM Neil Girdhar...