Add a workflow action to run the unit tests and a requirements.txt

PiperOrigin-RevId: 748454509

Author: qinyiyan

recml/.github/workflows/unit_tests.yaml

@@ -0,0 +1,24 @@
+name: Run Unit Tests
+
+on:
+  push:
+    branches: [ "main", "master" ]
+  pull_request:
+    branches: [ "main", "master" ]
+  workflow_dispatch:
+
+jobs:
+  run_tests:
+    name: Run Tests
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Install Dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt # If you have a requirements.txt file
+      - name: Run Pytest
+        run: |
+          export KERAS_BACKEND=jax
+          pytest -v

recml/requirements.txt

@@ -0,0 +1,17 @@
+jax>=0.4.30
+jaxlib>=0.4.30
+numpy
+pylint
+pytest
+pyink
+pre-commit
+clu
+jaxtyping
+tensorflow
+keras
+fiddle
+tensorflow_datasets
+absl-py
+keras_hub
+torch
+einops