From a4ef7905a3fec7513bb6640072e76ec398b0b30a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Olivier=20Can=C3=A9vet?= <olivier.canevet@idiap.ch>
Date: Thu, 12 Sep 2024 15:03:52 +0200
Subject: [PATCH] build: Add conda file for Jax backend

---
 conda/rsclf-jax.yaml | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 conda/rsclf-jax.yaml

diff --git a/conda/rsclf-jax.yaml b/conda/rsclf-jax.yaml
new file mode 100644
index 0000000..8416ea1
--- /dev/null
+++ b/conda/rsclf-jax.yaml
@@ -0,0 +1,16 @@
+# SPDX-FileCopyrightText: Copyright © Idiap Research Institute <contact@idiap.ch>
+#
+# SPDX-License-Identifier: GPL-3.0-only
+
+name: rsclf
+dependencies:
+  - python=3.11
+  - pip
+  - pip:
+    - keras
+    - h5py
+    - scipy
+    - jax[cuda12]
+    - matplotlib
+    - numpy
+    - polars