Brett Barkley issues

Repositories
Issues
Comments

Results 2 issues of


                                            Brett Barkley

Updates Needed in Datasets

The grad_updates_per_step hyperparameter for SAC changed at some point in the last year or two from being a decimal value to being an integer. I believe the old values are...

Persistent Caching of Jitted Functions on GPU for Brax Envs and Autodiff

Hi, I am trying to use persistent caching in XLA on GPU to speed up the execution of my Brax code. Tracking [issues](https://github.com/google/jax/discussions/13736#discussioncomment-5903999) in JAX this seems to be possible...