From 1c9cdc1fe29e2be1465b34f7c27b310e87e0de4a Mon Sep 17 00:00:00 2001
From: carlosgmartin <cgmartin@andrew.cmu.edu>
Date: Tue, 12 Mar 2024 18:34:43 -0400
Subject: [PATCH] Add ConvLSTMCell to docs.

---
 docs/api_reference/flax.linen/layers.rst | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docs/api_reference/flax.linen/layers.rst b/docs/api_reference/flax.linen/layers.rst
index b206c77c1..8232a084e 100644
--- a/docs/api_reference/flax.linen/layers.rst
+++ b/docs/api_reference/flax.linen/layers.rst
@@ -122,6 +122,10 @@ Recurrent
   :module: flax.linen
   :class: OptimizedLSTMCell
 
+.. flax_module::
+  :module: flax.linen
+  :class: ConvLSTMCell
+
 .. flax_module::
   :module: flax.linen
   :class: SimpleCell