Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .claude/skills/code-style/SKILL.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ Exception: `Dockerfile.ngc_pytorch` is exempt from this rule.

## Python Standard

Code must conform to Python 3.13.11+.
Code must conform to Python 3.13.13+.

## Indentation

Expand Down
2 changes: 1 addition & 1 deletion .python-version
Original file line number Diff line number Diff line change
@@ -1 +1 @@
3.13.11
3.13.13
4 changes: 2 additions & 2 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -69,8 +69,8 @@ RUN GITHUB_ARTIFACTORY=github.com \
&& rm -rf "${CMAKE_INSTALLER}" "${CMAKE_INSTALLER}.tar.gz"

# Install uv and python
ARG UV_VERSION=0.11.3
ARG PYTHON_VERSION=3.13.11
ARG UV_VERSION=0.11.6
ARG PYTHON_VERSION=3.13.13
ENV PATH="/root/.local/bin:$PATH"
RUN curl -LsSf https://astral.sh/uv/${UV_VERSION}/install.sh | sh && \
uv python install ${PYTHON_VERSION}
Expand Down
2 changes: 1 addition & 1 deletion docker/Dockerfile.ngc_pytorch
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ rm -rf /var/lib/apt/lists/*
EOF

# Install uv at /usr/local/bin in case the root home directory is bind mounted
ARG UV_VERSION=0.11.3
ARG UV_VERSION=0.11.6
RUN curl -LsSf https://astral.sh/uv/${UV_VERSION}/install.sh | XDG_BIN_HOME=/usr/local/bin sh

# Disable usage stats by default for users who are sensitive to sharing usage.
Expand Down
16 changes: 9 additions & 7 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ readme = { file = "README.md", content-type = "text/markdown" }
name = "nemo-rl"
dynamic = ["version", "readme"]
description = "NeMo RL: A Scalable and Efficient Post-Training Library for Models Ranging from 1 GPU to 1000s, and from Tiny to >100B Parameters"
requires-python = ">=3.13.11"
requires-python = ">=3.13.13"
license = { text = "Apache 2.0" }
dependencies = [
"setuptools",
Expand Down Expand Up @@ -181,9 +181,9 @@ triton = [
causal-conv1d = { git = "https://github.com/Dao-AILab/causal-conv1d", rev = "67e0a9dfe1518fc0036444e9ab5fe06ab78299e0" }
mamba-ssm = { git = "https://github.com/state-spaces/mamba.git", rev = "d68d16ed7d5d5164eb5a57c0285f3b7eb8394ec1" }
nv-grouped-gemm = { git = "https://github.com/fanshiqing/grouped_gemm", tag = "v1.1.4.post7" }
# From JustinTong0323/sglang branch update-transformers-v5 (sgl-project/sglang#17784)
sglang = { git = "https://github.com/JustinTong0323/sglang.git", rev = "70aa688742dd2b75bf9e8e980249303f39295b0d", subdirectory = "python" }
sgl-kernel = { git = "https://github.com/JustinTong0323/sglang.git", rev = "70aa688742dd2b75bf9e8e980249303f39295b0d", subdirectory = "sgl-kernel" }
# From chtruong814/sglang branch update-transformers-v5
sglang = { git = "https://github.com/chtruong814/sglang.git", rev = "78135940075cd87307f6a764d4402188d00ef7c3", subdirectory = "python" }
sgl-kernel = { git = "https://github.com/chtruong814/sglang.git", rev = "78135940075cd87307f6a764d4402188d00ef7c3", subdirectory = "sgl-kernel" }
emerging-optimizers = { git = "https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git", rev = "v0.2.0" }

[tool.uv.workspace]
Expand Down Expand Up @@ -270,6 +270,8 @@ override-dependencies = [
"mlflow>=3.9.0rc0",
# Override outlines for Python 3.13 support
"outlines>=0.2.0",
# Upgrade pytest to 9.0.3
"pytest>=9.0.3",
]
# CVE fixes
constraint-dependencies = [
Expand Down Expand Up @@ -391,10 +393,10 @@ requires-dist = ["torch", "scikit-build-core", "wheel"]

[[tool.uv.dependency-metadata]]
name = "sglang"
# VCS install from JustinTong0323/sglang@update-transformers-v5
# VCS install from chtruong814/sglang@update-transformers-v5
# Version is dynamic (setuptools-scm), so uv cannot resolve deps from the VCS source automatically.
# This requires-dist list must be kept in sync with the fork's python/pyproject.toml [project].dependencies.
# Source: https://github.com/JustinTong0323/sglang/blob/70aa688742dd2b75bf9e8e980249303f39295b0d/python/pyproject.toml
# Source: https://github.com/chtruong814/sglang/blob/78135940075cd87307f6a764d4402188d00ef7c3/python/pyproject.toml
version = "0.5.7.dev0"
requires-dist = [
"IPython",
Expand Down Expand Up @@ -450,7 +452,7 @@ requires-dist = [
"torchcodec==0.8.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')",
"torchvision",
"tqdm",
"transformers==5.3.0",
"transformers==5.2.0",
"uvicorn",
"uvloop",
"xgrammar==0.1.27",
Expand Down
2 changes: 1 addition & 1 deletion pyrefly.toml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
python-version = "3.13.11"
python-version = "3.13.13"
replace-imports-with-any = [
"nemo_automodel.*",
"pynvml.*",
Expand Down
2 changes: 1 addition & 1 deletion research/template_project/.python-version
Original file line number Diff line number Diff line change
@@ -1 +1 @@
3.13.11
3.13.13
2 changes: 1 addition & 1 deletion research/template_project/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ This command will:
## Python Version

> [!NOTE]
> This project uses Python 3.13.11 as specified in `.python-version`.
> This project uses Python 3.13.13 as specified in `.python-version`.
> This Python version should always be kept in sync with the `.python-version` file at the root of the `nemo-rl` repository to ensure compatibility.


Expand Down
2 changes: 1 addition & 1 deletion research/template_project/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ name = "template-project"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.13.11"
requires-python = ">=3.13.13"
dependencies = ["nemo-rl"]

[dependency-groups]
Expand Down
114 changes: 42 additions & 72 deletions uv.lock

Large diffs are not rendered by default.

Loading