seperate requirements

pyproject.toml

@@ -8,7 +8,6 @@ include = ['jaxmarl*']
 
 [tool.setuptools.dynamic]
 version = {attr = "jaxmarl.__version__"}
-dependencies = {file = ["requirements/requirements.txt"]}
 
 [project]
 name = "jaxmarl"
@@ -17,7 +16,7 @@ description = "Multi-Agent Reinforcement Learning with JAX"
 authors = [
     {name = "Foerster Lab for AI Research", email = "[email protected]"},
   ]
-dynamic = ["version", "dependencies"]
+dynamic = ["version"]
 license = {file = "LICENSE"}
 requires-python = ">=3.10"
 classifiers = [
@@ -31,6 +30,35 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
     "License :: OSI Approved :: Apache Software License",
 ]
+dependencies = [
+  "jax>=0.4.16.0,<=0.4.25",
+  "jaxlib>=0.4.16.0,<=0.4.25",
+  "flax",
+  "chex",
+  "brax==0.10.3",
+  "mujoco==3.1.3",
+  "matplotlib",
+  "pillow",
+  "scipy<=1.12",
+  "gymnax",
+]
+
+[project.optional-dependencies]
+alg = [
+  "optax",
+  "distrax",
+  "safetensors",
+  "flashbax==0.1.0",
+  "wandb",
+  "hydra-core>=1.3.2",
+  "omegaconf>=2.3.0",
+  "pettingzoo>=1.24.3",
+  "tqdm>=4.66.0",
+]
+dev = [
+  "pytest",
+  "pygame",
+]
 
 [project.urls]
 "Homepage" = "https://github.com/FLAIROx/JaxMARL"