From bad67fec378de4b645dc8642e425682fd153aa22 Mon Sep 17 00:00:00 2001
From: nathan <44041512+b0kch01@users.noreply.github.com>
Date: Tue, 30 Jan 2024 19:37:34 -0800
Subject: [PATCH] Added TeX line breaks to mlx.optimizers.Lion docstring (#595)

Fixes the "misplaced &" MathJax error in documentation.
---
 python/mlx/optimizers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/mlx/optimizers.py b/python/mlx/optimizers.py
index 1541b41cc..c47234bd5 100644
--- a/python/mlx/optimizers.py
+++ b/python/mlx/optimizers.py
@@ -465,8 +465,8 @@ class Lion(Optimizer):
 
     .. math::
 
-        c_{t + 1} &= \beta_1 m_t + (1 - \beta_1) g_t
-        m_{t + 1} &= \beta_2 m_t + (1 - \beta_2) g_t
+        c_{t + 1} &= \beta_1 m_t + (1 - \beta_1) g_t \\
+        m_{t + 1} &= \beta_2 m_t + (1 - \beta_2) g_t \\
         w_{t + 1} &= w_t - \eta (\text{sign}(c_t) + \lambda w_t)
 
     Args: