search

google / deluca

Performant, differentiable reinforcement learning
https://deluca.fyi
Apache License 2.0
121 stars 17 forks source link

dataclass fields that are int #42

Open danielsuo opened 3 years ago

danielsuo commented 3 years ago

Trainable because you don't want jax to think a changed value is a new pytree structure, but not trainable because int and jax don't play nice