NVIDIA · mgoldfarb-nvidia · Mar 23, 2026 · Mar 23, 2026 · Mar 23, 2026 · Mar 23, 2026
diff --git a/.github/container/Dockerfile.jax b/.github/container/Dockerfile.jax
@@ -134,6 +134,9 @@ ls ${SRC_PATH_TRANSFORMER_ENGINE}/dist/*.whl
 echo "transformer-engine @ file://$(ls ${SRC_PATH_TRANSFORMER_ENGINE}/dist/*.whl)" > /opt/pip-tools.d/requirements-te.in
 EOF
 
+## nvidia-cutlass-dsl
+RUN echo "nvidia-cutlass-dsl" >> /opt/pip-tools.d/requirements-cutlass-dsl.in
+
 ###############################################################################
 ## Install the nsys-jax JAX/XLA-aware profiling scripts, patch Nsight Systems
 ###############################################################################

diff --git a/.github/container/cutlass_dsl_jax/LICENSE b/.github/container/cutlass_dsl_jax/LICENSE
diff --git a/.github/container/cutlass_dsl_jax/README.md b/.github/container/cutlass_dsl_jax/README.md