Review the existing RNN implementation in Trax (e.g., trax/layers/rnn.py). Check the comments in the issue for discussion on how to add bidirectional support. Consider implementing a wrapper or modifying the RNN layer to accept a bidirectional flag. Look for any existing PRs or branches that attempt this.
