Skip to content

Update CuTe DSL JAX tutorial#3188

Open
katjasrz wants to merge 1 commit into
NVIDIA:mainfrom
katjasrz:update-cute-dsl-jax-tutorial
Open

Update CuTe DSL JAX tutorial#3188
katjasrz wants to merge 1 commit into
NVIDIA:mainfrom
katjasrz:update-cute-dsl-jax-tutorial

Conversation

@katjasrz
Copy link
Copy Markdown
Contributor

This PR syncs the CUTLASS tutorial with the improved version reviewed in JAX PR #36496, including the reviewer-driven fixes, cleaner explanations, updated JAX APIs.

@Junkai-Wu
Copy link
Copy Markdown
Collaborator

Thanks for the fix. We pushed out the 4.5 release lately and the dsl example folder structure has been reorganized. Please rebase the latest cutlass main branch and apply the changes to the correct file locations.

@katjasrz katjasrz force-pushed the update-cute-dsl-jax-tutorial branch from e4547eb to fcdadf8 Compare May 13, 2026 15:27
@katjasrz
Copy link
Copy Markdown
Contributor Author

@Junkai-Wu sure, I've rebased it — the structure is now the same as main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants