sokrypton / sokrypton/ColabFold
Nvidia RTX 6000 Blackwell / SM120 incompatibility
Open
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 2.9k
- Forks
- 747
- PR merge metrics
- No merged PRs in 30d
Description
Hello,
When trying to run Colabfold on RTX6000, it crashes as the version of jax you are using is incompatible with shaders version 12.
Alphafold3 resolved this issue in the head branch by updating jax/jaxlib to 0.8.x
Do you think it would be possible to fix this in Colabfold as well? We can gladly help with testing, if needed.
thanks!
k.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by locating where ColabFold pins or installs jax and jaxlib, then compare those versions with the update used by AlphaFold3's head branch. Test the updated setup on an Nvidia RTX 6000 with SM120; done means ColabFold no longer crashes from the shader-version incompatibility.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- bioinformatics, machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 55/100