Fix pallas GLU kernel on large inputs and update documentation for compilation buckets.
PiperOrigin-RevId: 696513320
parent
1d3e173b
No related branches found
No related tags found
Showing
- docs/known_issues.md 0 additions, 15 deletionsdocs/known_issues.md
- docs/performance.md 43 additions, 8 deletionsdocs/performance.md
- run_alphafold.py 13 additions, 17 deletionsrun_alphafold.py
- src/alphafold3/jax/common/array_view.py 36 additions, 24 deletionssrc/alphafold3/jax/common/array_view.py
- src/alphafold3/jax/gated_linear_unit/block.py 5 additions, 2 deletionssrc/alphafold3/jax/gated_linear_unit/block.py
- src/alphafold3/model/pipeline/pipeline.py 18 additions, 3 deletionssrc/alphafold3/model/pipeline/pipeline.py
Loading
Please register or sign in to comment