Skip to content

[codex] Document Tokamax RTX GPU compatibility#47

Merged
DimaMolod merged 1 commit into
mainfrom
codex/document-af3-tokamax-rtx
Jun 10, 2026
Merged

[codex] Document Tokamax RTX GPU compatibility#47
DimaMolod merged 1 commit into
mainfrom
codex/document-af3-tokamax-rtx

Conversation

@DimaMolod

@DimaMolod DimaMolod commented Jun 8, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Reword the slurm_exclude_nodes guidance so RTX PRO 6000 / Blackwell exclusion is documented as an old/pre-Tokamax container workaround.
  • Keep the exclusion mechanism documented for genuinely incompatible old images or local builds, but make clear that the permanent fix is the updated AlphaFold 3 / Tokamax stack.
  • Update the memory-resource test comment to match the new intent without changing behavior.

Context

The AlphaPulldown AF3 submodule update targets official AF3 v3.0.2, which includes the compatibility work requested in google-deepmind/alphafold3#394.

Official AF3 history used as evidence:

  • 3890782: migrated AF3 to tokamax.gated_linear_unit and tokamax.dot_product_attention.
  • 608edb6: updated to Tokamax 0.0.11 and JAX 0.9.1.
  • f6a5aec: official v3.0.2, containing both commits.

Coordinates with AlphaPulldown PR KosinskiLab/AlphaPulldown#623.

@DimaMolod DimaMolod marked this pull request as ready for review June 8, 2026 16:58
@DimaMolod DimaMolod merged commit 1958bd8 into main Jun 10, 2026
2 checks passed
@DimaMolod DimaMolod deleted the codex/document-af3-tokamax-rtx branch June 10, 2026 07:00
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.

1 participant