Allan Zhou comments

Repositories
Issues
Comments

Results 4 comments of


                                            Allan Zhou

Error running particle experiments

Hi Pete, thanks for the patience (just got back from various confs/travels). I verified the version numbers linked. Here's my entire pip list. Maybe a version update in some other...

Continual version of kitchen.

Ah apologies, I've been a bit busy for the past week. I'll try to do this within the next week.

Continual version of kitchen.

Thanks for the reminder, I've added a bunch of comments/documentation for the KitchenContinualEnv. Let me know if you have any questions!

Pallas+scan: NotImplementedError when num_extensive is True

I also tried a simpler implementation without using scan but just a loop, which produces a different error: ```python import jax from jax.experimental import pallas as pl import jax.numpy as...