Skip to content

feat: MLFlow Integration for experiment tracking #697

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 4 commits into from
Jul 21, 2025
Merged

feat: MLFlow Integration for experiment tracking #697

merged 4 commits into from
Jul 21, 2025

Conversation

terrykong
Copy link
Contributor

@terrykong terrykong commented Jul 20, 2025 •
edited
Loading

Copy of #534. Recreated to run CI + resolve conflicts

Closes #514

@therealnaveenkamal @terrykong
feat: MLFlow Integration for experiment tracking
accc73d 
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
@github-actions github-actions bot added the documentation Improvements or additions to documentation label Jul 20, 2025
@terrykong terrykong added the CI:L1 Run doctests, unit tests, and functional tests label Jul 20, 2025
@terrykong
not needed
8f24d9a 
Signed-off-by: Terry Kong <terryk@nvidia.com>
@terrykong terrykong added CI:L1 Run doctests, unit tests, and functional tests and removed CI:L1 Run doctests, unit tests, and functional tests labels Jul 20, 2025
terrykong added 2 commits July 20, 2025 22:41
@terrykong
fix the configs
f0a1990 
Signed-off-by: Terry Kong <terryk@nvidia.com>
@terrykong
add mlflow to top level configs
bf2d472 
Signed-off-by: Terry Kong <terryk@nvidia.com>
@terrykong terrykong added CI:L1 Run doctests, unit tests, and functional tests and removed CI:L1 Run doctests, unit tests, and functional tests labels Jul 21, 2025
@terrykong terrykong enabled auto-merge July 21, 2025 16:07
@terrykong terrykong added this pull request to the merge queue Jul 21, 2025
Merged via the queue into main with commit 9cbf9b4 Jul 21, 2025
21 of 23 checks passed
@terrykong terrykong deleted the pr-534-mlflow branch July 21, 2025 21:00
This was referenced Jul 21, 2025
Closed
Closed
jialei777 pushed a commit to jialei777/nemo-rl that referenced this pull request Jul 23, 2025
@terrykong @therealnaveenkamal @jialei777
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
75a5a6d 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 28, 2025
@terrykong @therealnaveenkamal @xxman-google
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
85f8a0d 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
KiddoZhu pushed a commit that referenced this pull request Jul 28, 2025
@terrykong @therealnaveenkamal @KiddoZhu
feat: MLFlow Integration for experiment tracking (#697)
cdb9f4c 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 30, 2025
@terrykong @therealnaveenkamal @xxman-google
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
3e171bd 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
@terrykong @therealnaveenkamal
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
da894a0 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
@terrykong @therealnaveenkamal @xxman-google
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
79e9d7f 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
soodoshll pushed a commit to soodoshll/RL that referenced this pull request Aug 13, 2025
@terrykong @therealnaveenkamal @soodoshll
feat: MLFlow Integration for experiment tracking (NVIDIA-NeMo#697)
f6033bb 
Signed-off-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Terry Kong <terryk@nvidia.com>
Co-authored-by: Naveenraj Kamalakannan <therealnaveenkamal@gmail.com>
Signed-off-by: Qidong Su <qidongs@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@parthchadha parthchadha parthchadha approved these changes

@SahilJain314 SahilJain314 Awaiting requested review from SahilJain314

Assignees
No one assigned
Labels
CI:L1 Run doctests, unit tests, and functional tests documentation Improvements or additions to documentation
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Support for MLflow
3 participants
@terrykong @parthchadha @therealnaveenkamal