Skip to content

recurrent

Module: recurrent.py

This module implements recurrent layers for Neural Networks

Authors
  • Lokesh Mohanty (lokeshm@iisc.ac.in)
Version Info
  • 06/01/2025: Initial version

GRUCell

Bases: GRUCell

Implements a GRU Cell

Source code in scirex/core/dl/nn/layers/recurrent.py 
41
42
43
44
class GRUCell(eqx.nn.GRUCell):
    """
    Implements a GRU Cell
    """

LSTMCell

Bases: LSTMCell

Implements a LSTM Cell

Source code in scirex/core/dl/nn/layers/recurrent.py 
47
48
49
50
class LSTMCell(eqx.nn.LSTMCell):
    """
    Implements a LSTM Cell
    """