parabellum 0.0.77__tar.gz → 0.0.79__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (11) hide show
  1. {parabellum-0.0.77 → parabellum-0.0.79}/PKG-INFO +4 -2
  2. {parabellum-0.0.77 → parabellum-0.0.79}/pyproject.toml +4 -4
  3. {parabellum-0.0.77 → parabellum-0.0.79}/uv.lock +159 -8
  4. {parabellum-0.0.77 → parabellum-0.0.79}/.gitignore +0 -0
  5. {parabellum-0.0.77 → parabellum-0.0.79}/README.md +0 -0
  6. {parabellum-0.0.77 → parabellum-0.0.79}/main.py +0 -0
  7. {parabellum-0.0.77 → parabellum-0.0.79}/parabellum/__init__.py +0 -0
  8. {parabellum-0.0.77 → parabellum-0.0.79}/parabellum/env.py +0 -0
  9. {parabellum-0.0.77 → parabellum-0.0.79}/parabellum/geo.py +0 -0
  10. {parabellum-0.0.77 → parabellum-0.0.79}/parabellum/types.py +0 -0
  11. {parabellum-0.0.77 → parabellum-0.0.79}/parabellum/utils.py +0 -0
{parabellum-0.0.77 → parabellum-0.0.79}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: parabellum
3
- Version: 0.0.77
3
+ Version: 0.0.79
4
4
  Summary: Parabellum environment for parallel warfare simulation
5
5
  Author-email: Noah Syrkis <desk@syrkis.com>
6
6
  Requires-Python: <3.12,>=3.11
@@ -14,7 +14,9 @@ Requires-Dist: flashbax<0.2,>=0.1.2
14
14
  Requires-Dist: flax<0.11,>=0.10.4
15
15
  Requires-Dist: geopy<3,>=2.4.1
16
16
  Requires-Dist: jax-tqdm<0.4,>=0.3.1
17
- Requires-Dist: jax>=0.6.0
17
+ Requires-Dist: jax>=0.6.0; sys_platform == 'win32'
18
+ Requires-Dist: jax[cpu]>=0.6.0; sys_platform == 'darwin'
19
+ Requires-Dist: jax[cuda12]>=0.6.0; sys_platform == 'linux'
18
20
  Requires-Dist: jaxkd>=0.1.0
19
21
  Requires-Dist: jaxtyping<0.3,>=0.2.33
20
22
  Requires-Dist: navix<0.8,>=0.7.0
{parabellum-0.0.77 → parabellum-0.0.79}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "parabellum"
3
- version = "0.0.77"
3
+ version = "0.0.79"
4
4
  description = "Parabellum environment for parallel warfare simulation"
5
5
  authors = [{ name = "Noah Syrkis", email = "desk@syrkis.com" }]
6
6
  requires-python = ">=3.11,<3.12"
@@ -25,14 +25,14 @@ dependencies = [
25
25
  "jaxkd>=0.1.0",
26
26
  "chex>=0.1.89",
27
27
  "osmnx>=2.0.4",
28
- "jax>=0.6.0",
28
+ "jax[cpu]>=0.6.0; sys_platform == 'darwin'",
29
+ "jax[cuda12]>=0.6.0; sys_platform == 'linux'",
30
+ "jax>=0.6.0; sys_platform == 'win32'",
29
31
  ]
30
32
 
31
33
  [dependency-groups]
32
34
  dev = ["esch"]
33
35
 
34
- [tool.uv]
35
-
36
36
  [tool.uv.sources]
37
37
  esch = { path = "../../esch" }
38
38
 
{parabellum-0.0.77 → parabellum-0.0.79}/uv.lock RENAMED
@@ -3,7 +3,9 @@ revision = 2
3
3
  requires-python = "==3.11.*"
4
4
  resolution-markers = [
5
5
  "sys_platform == 'linux'",
6
- "sys_platform != 'linux'",
6
+ "sys_platform == 'darwin'",
7
+ "sys_platform == 'win32'",
8
+ "sys_platform != 'darwin' and sys_platform != 'linux' and sys_platform != 'win32'",
7
9
  ]
8
10
 
9
11
  [[package]]
@@ -997,6 +999,47 @@ wheels = [
997
999
  { url = "https://files.pythonhosted.org/packages/31/25/32c5e2c919da4faaea9ef5088437ab6e01738c49402e4ec8a6c7b49e30ef/jax-0.6.0-py3-none-any.whl", hash = "sha256:22b21827597c6d6b46e88543b4fc372fcddf1cc1247660452de020cc4bda1afc", size = 2338416, upload-time = "2025-04-17T00:00:18.542Z" },
998
1000
  ]
999
1001
 
1002
+ [package.optional-dependencies]
1003
+ cuda12 = [
1004
+ { name = "jax-cuda12-plugin", extra = ["with-cuda"], marker = "sys_platform == 'linux'" },
1005
+ { name = "jaxlib", marker = "sys_platform == 'linux'" },
1006
+ ]
1007
+
1008
+ [[package]]
1009
+ name = "jax-cuda12-pjrt"
1010
+ version = "0.6.0"
1011
+ source = { registry = "https://pypi.org/simple" }
1012
+ wheels = [
1013
+ { url = "https://files.pythonhosted.org/packages/8c/38/5e9f0da742fa4d9ce8d4bcb6e2f2cb1f0546a428e1a787ceba4b2569d421/jax_cuda12_pjrt-0.6.0-py3-none-manylinux2014_aarch64.whl", hash = "sha256:9bfebb06a39614cb6899f7730ea8561f11156ac81cbb3ec6884a62afb3b15ff3", size = 109541080, upload-time = "2025-04-17T00:00:22.384Z" },
1014
+ { url = "https://files.pythonhosted.org/packages/18/9f/8b83780974bc3ac3ef2481f6abbff86e4a55a83d52dbbbcce732de15cf18/jax_cuda12_pjrt-0.6.0-py3-none-manylinux2014_x86_64.whl", hash = "sha256:68371bd9c135244b89663039be208255698a75bec9854d419ea3c3f957ca4646", size = 123355777, upload-time = "2025-04-17T00:00:28.646Z" },
1015
+ ]
1016
+
1017
+ [[package]]
1018
+ name = "jax-cuda12-plugin"
1019
+ version = "0.6.0"
1020
+ source = { registry = "https://pypi.org/simple" }
1021
+ dependencies = [
1022
+ { name = "jax-cuda12-pjrt", marker = "sys_platform == 'linux'" },
1023
+ ]
1024
+ wheels = [
1025
+ { url = "https://files.pythonhosted.org/packages/b0/db/8acca1de61d9e574756930246c672812cb0cd37c607b4bc865c5250c7423/jax_cuda12_plugin-0.6.0-cp311-cp311-manylinux2014_aarch64.whl", hash = "sha256:7cd1b488a54a3089e89588ccaf677089952c82529e7d0403e0b050199e525418", size = 15476646, upload-time = "2025-04-17T00:00:37.955Z" },
1026
+ { url = "https://files.pythonhosted.org/packages/74/ce/2aecd778f8ae0d08194110a912d5b87a2569e9c8e7b97cb46a2c6aa2678a/jax_cuda12_plugin-0.6.0-cp311-cp311-manylinux2014_x86_64.whl", hash = "sha256:0d9ecede66c40258702a42261e868cdb56a103551a7c3c884b35f531c9acd48e", size = 15786927, upload-time = "2025-04-17T00:00:40.081Z" },
1027
+ ]
1028
+
1029
+ [package.optional-dependencies]
1030
+ with-cuda = [
1031
+ { name = "nvidia-cublas-cu12", marker = "sys_platform == 'linux'" },
1032
+ { name = "nvidia-cuda-cupti-cu12", marker = "sys_platform == 'linux'" },
1033
+ { name = "nvidia-cuda-nvcc-cu12", marker = "sys_platform == 'linux'" },
1034
+ { name = "nvidia-cuda-runtime-cu12", marker = "sys_platform == 'linux'" },
1035
+ { name = "nvidia-cudnn-cu12", marker = "sys_platform == 'linux'" },
1036
+ { name = "nvidia-cufft-cu12", marker = "sys_platform == 'linux'" },
1037
+ { name = "nvidia-cusolver-cu12", marker = "sys_platform == 'linux'" },
1038
+ { name = "nvidia-cusparse-cu12", marker = "sys_platform == 'linux'" },
1039
+ { name = "nvidia-nccl-cu12", marker = "sys_platform == 'linux'" },
1040
+ { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux'" },
1041
+ ]
1042
+
1000
1043
  [[package]]
1001
1044
  name = "jax-tqdm"
1002
1045
  version = "0.3.1"
@@ -1219,7 +1262,7 @@ dependencies = [
1219
1262
  { name = "overrides" },
1220
1263
  { name = "packaging" },
1221
1264
  { name = "prometheus-client" },
1222
- { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'linux'" },
1265
+ { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'darwin' and sys_platform != 'linux'" },
1223
1266
  { name = "pyzmq" },
1224
1267
  { name = "send2trash" },
1225
1268
  { name = "terminado" },
@@ -1237,7 +1280,7 @@ name = "jupyter-server-terminals"
1237
1280
  version = "0.5.3"
1238
1281
  source = { registry = "https://pypi.org/simple" }
1239
1282
  dependencies = [
1240
- { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'linux'" },
1283
+ { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'darwin' and sys_platform != 'linux'" },
1241
1284
  { name = "terminado" },
1242
1285
  ]
1243
1286
  sdist = { url = "https://files.pythonhosted.org/packages/fc/d5/562469734f476159e99a55426d697cbf8e7eb5efe89fb0e0b4f83a3d3459/jupyter_server_terminals-0.5.3.tar.gz", hash = "sha256:5ae0295167220e9ace0edcfdb212afd2b01ee8d179fe6f23c899590e9b8a5269", size = 31430, upload-time = "2024-03-12T14:37:03.049Z" }
@@ -1628,6 +1671,110 @@ wheels = [
1628
1671
  { url = "https://files.pythonhosted.org/packages/1e/48/a9a4b538e28f854bfb62e1dea3c8fea12e90216a276c7777ae5345ff29a7/numpy-2.1.3-cp311-cp311-win_amd64.whl", hash = "sha256:d89dd2b6da69c4fff5e39c28a382199ddedc3a5be5390115608345dec660b9e2", size = 12869487, upload-time = "2024-11-02T17:36:52.909Z" },
1629
1672
  ]
1630
1673
 
1674
+ [[package]]
1675
+ name = "nvidia-cublas-cu12"
1676
+ version = "12.9.1.4"
1677
+ source = { registry = "https://pypi.org/simple" }
1678
+ wheels = [
1679
+ { url = "https://files.pythonhosted.org/packages/82/6c/90d3f532f608a03a13c1d6c16c266ffa3828e8011b1549d3b61db2ad59f5/nvidia_cublas_cu12-12.9.1.4-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:7a950dae01add3b415a5a5cdc4ec818fb5858263e9cca59004bb99fdbbd3a5d6", size = 575006342, upload-time = "2025-06-05T20:04:16.902Z" },
1680
+ { url = "https://files.pythonhosted.org/packages/77/3c/aa88abe01f3be3d1f8f787d1d33dc83e76fec05945f9a28fbb41cfb99cd5/nvidia_cublas_cu12-12.9.1.4-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:453611eb21a7c1f2c2156ed9f3a45b691deda0440ec550860290dc901af5b4c2", size = 581242350, upload-time = "2025-06-05T20:04:51.979Z" },
1681
+ ]
1682
+
1683
+ [[package]]
1684
+ name = "nvidia-cuda-cupti-cu12"
1685
+ version = "12.9.79"
1686
+ source = { registry = "https://pypi.org/simple" }
1687
+ wheels = [
1688
+ { url = "https://files.pythonhosted.org/packages/b4/78/351b5c8cdbd9a6b4fb0d6ee73fb176dcdc1b6b6ad47c2ffff5ae8ca4a1f7/nvidia_cuda_cupti_cu12-12.9.79-py3-none-manylinux_2_25_aarch64.whl", hash = "sha256:791853b030602c6a11d08b5578edfb957cadea06e9d3b26adbf8d036135a4afe", size = 10077166, upload-time = "2025-06-05T20:01:01.385Z" },
1689
+ { url = "https://files.pythonhosted.org/packages/c1/2e/b84e32197e33f39907b455b83395a017e697c07a449a2b15fd07fc1c9981/nvidia_cuda_cupti_cu12-12.9.79-py3-none-manylinux_2_25_x86_64.whl", hash = "sha256:096bcf334f13e1984ba36685ad4c1d6347db214de03dbb6eebb237b41d9d934f", size = 10814997, upload-time = "2025-06-05T20:01:10.168Z" },
1690
+ ]
1691
+
1692
+ [[package]]
1693
+ name = "nvidia-cuda-nvcc-cu12"
1694
+ version = "12.9.86"
1695
+ source = { registry = "https://pypi.org/simple" }
1696
+ wheels = [
1697
+ { url = "https://files.pythonhosted.org/packages/25/48/b54a06168a2190572a312bfe4ce443687773eb61367ced31e064953dd2f7/nvidia_cuda_nvcc_cu12-12.9.86-py3-none-manylinux2010_x86_64.manylinux_2_12_x86_64.whl", hash = "sha256:5d6a0d32fdc7ea39917c20065614ae93add6f577d840233237ff08e9a38f58f0", size = 40546229, upload-time = "2025-06-05T20:01:53.357Z" },
1698
+ { url = "https://files.pythonhosted.org/packages/d6/5c/8cc072436787104bbbcbde1f76ab4a0d89e68f7cebc758dd2ad7913a43d0/nvidia_cuda_nvcc_cu12-12.9.86-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:44e1eca4d08926193a558d2434b1bf83d57b4d5743e0c431c0c83d51da1df62b", size = 39411138, upload-time = "2025-06-05T20:01:43.182Z" },
1699
+ ]
1700
+
1701
+ [[package]]
1702
+ name = "nvidia-cuda-runtime-cu12"
1703
+ version = "12.9.79"
1704
+ source = { registry = "https://pypi.org/simple" }
1705
+ wheels = [
1706
+ { url = "https://files.pythonhosted.org/packages/bc/e0/0279bd94539fda525e0c8538db29b72a5a8495b0c12173113471d28bce78/nvidia_cuda_runtime_cu12-12.9.79-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:83469a846206f2a733db0c42e223589ab62fd2fabac4432d2f8802de4bded0a4", size = 3515012, upload-time = "2025-06-05T20:00:35.519Z" },
1707
+ { url = "https://files.pythonhosted.org/packages/bc/46/a92db19b8309581092a3add7e6fceb4c301a3fd233969856a8cbf042cd3c/nvidia_cuda_runtime_cu12-12.9.79-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:25bba2dfb01d48a9b59ca474a1ac43c6ebf7011f1b0b8cc44f54eb6ac48a96c3", size = 3493179, upload-time = "2025-06-05T20:00:53.735Z" },
1708
+ ]
1709
+
1710
+ [[package]]
1711
+ name = "nvidia-cudnn-cu12"
1712
+ version = "9.10.2.21"
1713
+ source = { registry = "https://pypi.org/simple" }
1714
+ dependencies = [
1715
+ { name = "nvidia-cublas-cu12", marker = "sys_platform == 'linux'" },
1716
+ ]
1717
+ wheels = [
1718
+ { url = "https://files.pythonhosted.org/packages/fa/41/e79269ce215c857c935fd86bcfe91a451a584dfc27f1e068f568b9ad1ab7/nvidia_cudnn_cu12-9.10.2.21-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:c9132cc3f8958447b4910a1720036d9eff5928cc3179b0a51fb6d167c6cc87d8", size = 705026878, upload-time = "2025-06-06T21:52:51.348Z" },
1719
+ { url = "https://files.pythonhosted.org/packages/ba/51/e123d997aa098c61d029f76663dedbfb9bc8dcf8c60cbd6adbe42f76d049/nvidia_cudnn_cu12-9.10.2.21-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:949452be657fa16687d0930933f032835951ef0892b37d2d53824d1a84dc97a8", size = 706758467, upload-time = "2025-06-06T21:54:08.597Z" },
1720
+ ]
1721
+
1722
+ [[package]]
1723
+ name = "nvidia-cufft-cu12"
1724
+ version = "11.4.1.4"
1725
+ source = { registry = "https://pypi.org/simple" }
1726
+ dependencies = [
1727
+ { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux'" },
1728
+ ]
1729
+ wheels = [
1730
+ { url = "https://files.pythonhosted.org/packages/9b/2b/76445b0af890da61b501fde30650a1a4bd910607261b209cccb5235d3daa/nvidia_cufft_cu12-11.4.1.4-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:1a28c9b12260a1aa7a8fd12f5ebd82d027963d635ba82ff39a1acfa7c4c0fbcf", size = 200822453, upload-time = "2025-06-05T20:05:27.889Z" },
1731
+ { url = "https://files.pythonhosted.org/packages/95/f4/61e6996dd20481ee834f57a8e9dca28b1869366a135e0d42e2aa8493bdd4/nvidia_cufft_cu12-11.4.1.4-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:c67884f2a7d276b4b80eb56a79322a95df592ae5e765cf1243693365ccab4e28", size = 200877592, upload-time = "2025-06-05T20:05:45.862Z" },
1732
+ ]
1733
+
1734
+ [[package]]
1735
+ name = "nvidia-cusolver-cu12"
1736
+ version = "11.7.5.82"
1737
+ source = { registry = "https://pypi.org/simple" }
1738
+ dependencies = [
1739
+ { name = "nvidia-cublas-cu12", marker = "sys_platform == 'linux'" },
1740
+ { name = "nvidia-cusparse-cu12", marker = "sys_platform == 'linux'" },
1741
+ { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux'" },
1742
+ ]
1743
+ wheels = [
1744
+ { url = "https://files.pythonhosted.org/packages/03/99/686ff9bf3a82a531c62b1a5c614476e8dfa24a9d89067aeedf3592ee4538/nvidia_cusolver_cu12-11.7.5.82-py3-none-manylinux_2_27_aarch64.whl", hash = "sha256:62efa83e4ace59a4c734d052bb72158e888aa7b770e1a5f601682f16fe5b4fd2", size = 337869834, upload-time = "2025-06-05T20:06:53.125Z" },
1745
+ { url = "https://files.pythonhosted.org/packages/33/40/79b0c64d44d6c166c0964ec1d803d067f4a145cca23e23925fd351d0e642/nvidia_cusolver_cu12-11.7.5.82-py3-none-manylinux_2_27_x86_64.whl", hash = "sha256:15da72d1340d29b5b3cf3fd100e3cd53421dde36002eda6ed93811af63c40d88", size = 338117415, upload-time = "2025-06-05T20:07:16.809Z" },
1746
+ ]
1747
+
1748
+ [[package]]
1749
+ name = "nvidia-cusparse-cu12"
1750
+ version = "12.5.10.65"
1751
+ source = { registry = "https://pypi.org/simple" }
1752
+ dependencies = [
1753
+ { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux'" },
1754
+ ]
1755
+ wheels = [
1756
+ { url = "https://files.pythonhosted.org/packages/5e/6f/8710fbd17cdd1d0fc3fea7d36d5b65ce1933611c31e1861da330206b253a/nvidia_cusparse_cu12-12.5.10.65-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:221c73e7482dd93eda44e65ce567c031c07e2f93f6fa0ecd3ba876a195023e83", size = 366359408, upload-time = "2025-06-05T20:07:42.501Z" },
1757
+ { url = "https://files.pythonhosted.org/packages/12/46/b0fd4b04f86577921feb97d8e2cf028afe04f614d17fb5013de9282c9216/nvidia_cusparse_cu12-12.5.10.65-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:73060ce019ac064a057267c585bf1fd5a353734151f87472ff02b2c5c9984e78", size = 366465088, upload-time = "2025-06-05T20:08:20.413Z" },
1758
+ ]
1759
+
1760
+ [[package]]
1761
+ name = "nvidia-nccl-cu12"
1762
+ version = "2.27.5"
1763
+ source = { registry = "https://pypi.org/simple" }
1764
+ wheels = [
1765
+ { url = "https://files.pythonhosted.org/packages/bb/1c/857979db0ef194ca5e21478a0612bcdbbe59458d7694361882279947b349/nvidia_nccl_cu12-2.27.5-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:31432ad4d1fb1004eb0c56203dc9bc2178a1ba69d1d9e02d64a6938ab5e40e7a", size = 322400625, upload-time = "2025-06-26T04:11:04.496Z" },
1766
+ { url = "https://files.pythonhosted.org/packages/6e/89/f7a07dc961b60645dbbf42e80f2bc85ade7feb9a491b11a1e973aa00071f/nvidia_nccl_cu12-2.27.5-py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl", hash = "sha256:ad730cf15cb5d25fe849c6e6ca9eb5b76db16a80f13f425ac68d8e2e55624457", size = 322348229, upload-time = "2025-06-26T04:11:28.385Z" },
1767
+ ]
1768
+
1769
+ [[package]]
1770
+ name = "nvidia-nvjitlink-cu12"
1771
+ version = "12.9.86"
1772
+ source = { registry = "https://pypi.org/simple" }
1773
+ wheels = [
1774
+ { url = "https://files.pythonhosted.org/packages/46/0c/c75bbfb967457a0b7670b8ad267bfc4fffdf341c074e0a80db06c24ccfd4/nvidia_nvjitlink_cu12-12.9.86-py3-none-manylinux2010_x86_64.manylinux_2_12_x86_64.whl", hash = "sha256:e3f1171dbdc83c5932a45f0f4c99180a70de9bd2718c1ab77d14104f6d7147f9", size = 39748338, upload-time = "2025-06-05T20:10:25.613Z" },
1775
+ { url = "https://files.pythonhosted.org/packages/97/bc/2dcba8e70cf3115b400fef54f213bcd6715a3195eba000f8330f11e40c45/nvidia_nvjitlink_cu12-12.9.86-py3-none-manylinux2014_aarch64.manylinux_2_17_aarch64.whl", hash = "sha256:994a05ef08ef4b0b299829cde613a424382aff7efb08a7172c1fa616cc3af2ca", size = 39514880, upload-time = "2025-06-05T20:10:04.89Z" },
1776
+ ]
1777
+
1631
1778
  [[package]]
1632
1779
  name = "omegaconf"
1633
1780
  version = "2.3.0"
@@ -1757,7 +1904,7 @@ wheels = [
1757
1904
 
1758
1905
  [[package]]
1759
1906
  name = "parabellum"
1760
- version = "0.0.76"
1907
+ version = "0.0.79"
1761
1908
  source = { editable = "." }
1762
1909
  dependencies = [
1763
1910
  { name = "cachier" },
@@ -1769,7 +1916,9 @@ dependencies = [
1769
1916
  { name = "flashbax" },
1770
1917
  { name = "flax" },
1771
1918
  { name = "geopy" },
1772
- { name = "jax" },
1919
+ { name = "jax", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" },
1920
+ { name = "jax", marker = "sys_platform == 'darwin'" },
1921
+ { name = "jax", extra = ["cuda12"], marker = "sys_platform == 'linux'" },
1773
1922
  { name = "jax-tqdm" },
1774
1923
  { name = "jaxkd" },
1775
1924
  { name = "jaxtyping" },
@@ -1798,7 +1947,9 @@ requires-dist = [
1798
1947
  { name = "flashbax", specifier = ">=0.1.2,<0.2" },
1799
1948
  { name = "flax", specifier = ">=0.10.4,<0.11" },
1800
1949
  { name = "geopy", specifier = ">=2.4.1,<3" },
1801
- { name = "jax", specifier = ">=0.6.0" },
1950
+ { name = "jax", marker = "sys_platform == 'win32'", specifier = ">=0.6.0" },
1951
+ { name = "jax", extras = ["cpu"], marker = "sys_platform == 'darwin'", specifier = ">=0.6.0" },
1952
+ { name = "jax", extras = ["cuda12"], marker = "sys_platform == 'linux'", specifier = ">=0.6.0" },
1802
1953
  { name = "jax-tqdm", specifier = ">=0.3.1,<0.4" },
1803
1954
  { name = "jaxkd", specifier = ">=0.1.0" },
1804
1955
  { name = "jaxtyping", specifier = ">=0.2.33,<0.3" },
@@ -1849,7 +2000,7 @@ name = "pexpect"
1849
2000
  version = "4.9.0"
1850
2001
  source = { registry = "https://pypi.org/simple" }
1851
2002
  dependencies = [
1852
- { name = "ptyprocess" },
2003
+ { name = "ptyprocess", marker = "sys_platform != 'win32'" },
1853
2004
  ]
1854
2005
  sdist = { url = "https://files.pythonhosted.org/packages/42/92/cc564bf6381ff43ce1f4d06852fc19a2f11d180f23dc32d9588bee2f149d/pexpect-4.9.0.tar.gz", hash = "sha256:ee7d41123f3c9911050ea2c2dac107568dc43b2d3b0c7557a33212c398ead30f", size = 166450, upload-time = "2023-11-25T09:07:26.339Z" }
1855
2006
  wheels = [
@@ -2654,7 +2805,7 @@ version = "0.18.1"
2654
2805
  source = { registry = "https://pypi.org/simple" }
2655
2806
  dependencies = [
2656
2807
  { name = "ptyprocess", marker = "os_name != 'nt'" },
2657
- { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'linux'" },
2808
+ { name = "pywinpty", marker = "os_name == 'nt' and sys_platform != 'darwin' and sys_platform != 'linux'" },
2658
2809
  { name = "tornado" },
2659
2810
  ]
2660
2811
  sdist = { url = "https://files.pythonhosted.org/packages/8a/11/965c6fd8e5cc254f1fe142d547387da17a8ebfd75a3455f637c663fb38a0/terminado-0.18.1.tar.gz", hash = "sha256:de09f2c4b85de4765f7714688fff57d3e75bad1f909b589fde880460c753fd2e", size = 32701, upload-time = "2024-03-12T14:34:39.026Z" }
{parabellum-0.0.77 → parabellum-0.0.79}/.gitignore RENAMED
File without changes
{parabellum-0.0.77 → parabellum-0.0.79}/README.md RENAMED
File without changes
{parabellum-0.0.77 → parabellum-0.0.79}/main.py RENAMED
File without changes