diff --git a/cmd/runtimes/mlx/requirements.txt b/cmd/runtimes/mlx/requirements.txt
index 22ca51f32..87fc8a821 100644
--- a/cmd/runtimes/mlx/requirements.txt
+++ b/cmd/runtimes/mlx/requirements.txt
@@ -1,6 +1,6 @@
 # MLX libraries.
 mlx[cuda]==0.30.0
 mlx-data==0.1.0
-mlx-lm==0.28.3
+mlx-lm==0.31.1
 # HuggingFace libraries.
 datasets==4.4.1