Flax is changing the `RNNCellBase` API:

google / deluca

Performant, differentiable reinforcement learning

https://deluca.fyi

Apache License 2.0

121 stars 17 forks source link

Open copybara-service[bot] opened 1 year ago

copybara-service[bot] commented 1 year ago

Flax is changing the RNNCellBase API:

when calling the constructor of a class, it is now required to pass in a features argument
when calling the initialize_carry method, instead of passing in the batch_dims and size, you only have to pass in an input_shape

More details about the changes and how to upgrade to the new API can be found here.