More documentation progress.

vbharadwaj-bk · vbharadwaj-bk · commit 9d6e30e9e0f4 · 2025-12-24T19:23:27.000-08:00
diff --git a/docs/tests_and_benchmarks.rst b/docs/tests_and_benchmarks.rst
@@ -32,6 +32,14 @@ To set up the editable install and run the entire testsuite, use:
 
 Browse the ``tests`` directory to run specific components. 
 
+To test the JAX wrappers, follow the same steps above and make sure that
+``openequivariance_extjax`` is installed without build isolation. Then run
+
+.. code-block:: bash
+
+    pytest --jax tests/batch_test.py
+    pytest --jax tests/conv_test.py
+
 
 Replicating our Benchmarks
 ------------------------------
diff --git a/openequivariance/pyproject.toml b/openequivariance/pyproject.toml
@@ -65,6 +65,8 @@ dev = [
   "furo",
   "sphinx",
   "sphinx-autobuild"
+  "nanobind",
+  "scikit-build-core"
 ]
 
 [tool.setuptools.packages.find]

Original file line number	Diff line number	Diff line change
`@@ -65,6 +65,8 @@ dev = [`
`65`	`65`	`"furo",`
`66`	`66`	`"sphinx",`
`67`	`67`	`"sphinx-autobuild"`
	`68`	`+ "nanobind",`
	`69`	`+ "scikit-build-core"`
`68`	`70`	`]`
`69`	`71`
`70`	`72`	`[tool.setuptools.packages.find]`