From 9e44992053d7a21b0e814acda8fea116eb3fd30a Mon Sep 17 00:00:00 2001
From: Silia Taider <siliataider@gmail.com>
Date: Tue, 9 Jun 2026 18:26:38 +0200
Subject: [PATCH] [ML] Add jax dependency needed for data loading tests

---
 requirements.txt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 7b114fd9f5cb9..12acf702b324a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -17,6 +17,9 @@ tensorflow ; python_version < "3.14"
 torch
 xgboost
 
+# RDataLoader
+jax
+
 # PyROOT: ROOT.Numba.Declare decorator
 numba
 cffi