hookeai.model_architectures.rnn_base_model.custom

Model: Custom VMAP compatible GRU architecture.

Modules

hookeai.model_architectures.rnn_base_model.custom.gru_vmap

Multi-layer gated recurrent unit (GRU) recurrent neural network model.