Merge pull request #21 from MilagrosMarin/update_docs

chore: split `kpms` into separate optional dependencies for CPU and GPU, update README

Author: ttngu207

README.md

@@ -22,10 +22,18 @@ DataJoint Elements collectively standardize and automate data collection and ana
   git clone https://github.com/<enter_github_username>/element-moseq.git
   ```
 
++ Install [Conda](https://docs.conda.io/en/latest/miniconda.html).
+
++ Create and activate a Conda environment:
+  ```bash
+   conda env create -f conda_env.yml
+   conda activate element_moseq
+   ```
+
 + Install with `pip`:
 
   ```bash
-  pip install -e .[keypoint-moseq]
+  pip install -e .[kpms_cpu]  # MacOS: pip install -e ".[kpms_cpu]"
   ```
 
 + [Interactive tutorial on GitHub Codespaces](https://github.com/datajoint/element-moseq#interactive-tutorial)

pyproject.toml

@@ -34,7 +34,9 @@ elements = [
     "element-interface @ git+https://github.com/datajoint/element-interface.git",
     "element-animal @ git+https://github.com/datajoint/element-animal.git",
 ]
-kpms = ["jax<0.7.0", "keypoint-moseq @ git+https://github.com/dattalab/keypoint-moseq.git"]
+kpms_cpu = ["jax<0.7.0", "keypoint-moseq @ git+https://github.com/dattalab/keypoint-moseq.git"]
+kpms_gpu = ["jax<0.7.0", "keypoint-moseq[cuda] @ git+https://github.com/dattalab/keypoint-moseq.git"]
+
 tests = [
     "pytest",
     "pytest-cov",