Merge branch 'main' into imdb

Author: rasbt

.github/workflows/basic-tests-old-pytorch.yml

@@ -1,4 +1,4 @@
-name: Test PyTorch 2.2 and 2.6
+name: Test PyTorch 2.3 and 2.5
 
 on:
   push:
@@ -23,7 +23,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        pytorch-version: [ 2.3.0, 2.6.0 ]
+        pytorch-version: [ 2.3.0, 2.5.0 ]
 
     steps:
     - uses: actions/checkout@v4

pyproject.toml

@@ -9,18 +9,22 @@ description = "Implement a ChatGPT-like LLM in PyTorch from scratch, step by ste
 readme = "README.md"
 requires-python = ">=3.10,<=3.13"
 dependencies = [
-    "torch>=2.2.2,<2.6; sys_platform == 'darwin' and platform_machine == 'x86_64'",    # Intel macOS
-    "torch>=2.3.0; sys_platform != 'darwin' or platform_machine != 'x86_64'",
-    "jupyterlab>=4.0",
-    "tiktoken>=0.5.1",
-    "matplotlib>=3.7.1",
-    "tensorflow>=2.16.2; sys_platform == 'darwin' and platform_machine == 'x86_64'",  # Intel macOS
-    "tensorflow>=2.18.0; sys_platform != 'darwin' or platform_machine != 'x86_64'",
-    "tqdm>=4.66.1",
-    "numpy>=1.26,<2.1",
-    "pandas>=2.2.1",
-    "pip>=25.0.1",
-    "pytest>=8.3.5",
+  "torch>=2.2.2,<2.6; sys_platform == 'darwin' and platform_machine == 'x86_64' and python_version < '3.13'",
+  "torch>=2.2.2; sys_platform == 'darwin' and platform_machine == 'arm64'",
+  "torch>=2.2.2; sys_platform == 'linux'",
+  "torch>=2.2.2; sys_platform == 'win32'",
+  "tensorflow>=2.16.2; sys_platform == 'darwin' and platform_machine == 'x86_64'",
+  "tensorflow>=2.18.0; sys_platform == 'darwin' and platform_machine == 'arm64'",
+  "tensorflow>=2.18.0; sys_platform == 'linux'",
+  "tensorflow>=2.18.0; sys_platform == 'win32'",
+  "jupyterlab>=4.0",
+  "tiktoken>=0.5.1",
+  "matplotlib>=3.7.1",
+  "tqdm>=4.66.1",
+  "numpy>=1.26,<2.1",
+  "pandas>=2.2.1",
+  "pip>=25.0.1",
+  "pytest>=8.3.5",
 ]
 
 [tool.uv.sources]

requirements.txt

@@ -1,5 +1,5 @@
 torch>=2.2.2,<2.6; sys_platform == 'darwin' and platform_machine == 'x86_64'  # Intel macOS
-torch >= 2.3.0; sys_platform != 'darwin' or platform_machine != 'x86_64'   # all chapters
+torch >= 2.2.2; sys_platform != 'darwin' or platform_machine != 'x86_64'   # all chapters
 jupyterlab >= 4.0          # all
 tiktoken >= 0.5.1          # ch02; ch04; ch05
 matplotlib >= 3.7.1        # ch04; ch06; ch07