build: Add conda file for Jax backend

idiap · Sep 12, 2024 · a4ef790 · a4ef790
1 parent c7c9178
commit a4ef790
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/conda/rsclf-jax.yaml b/conda/rsclf-jax.yaml
@@ -0,0 +1,16 @@
+# SPDX-FileCopyrightText: Copyright © Idiap Research Institute <[email protected]>
+#
+# SPDX-License-Identifier: GPL-3.0-only
+
+name: rsclf
+dependencies:
+  - python=3.11
+  - pip
+  - pip:
+    - keras
+    - h5py
+    - scipy
+    - jax[cuda12]
+    - matplotlib
+    - numpy
+    - polars