Skip to content

Pull requests: AI-Hypercomputer/maxdiffusion

New pull request New
21 Open 226 Closed
21 Open 226 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Flax
#264 opened Oct 9, 2025 by susanbao Loading…
1
Add logical axis to mesh mapping in init
#263 opened Oct 9, 2025 by coolkp Loading…
1
Fix Unit test failure for JAX/Flax version update
#262 opened Oct 7, 2025 by susanbao Loading…
4
Jax 0.7.2 SDXL Workload OOM
#260 opened Oct 6, 2025 by susanbao Loading…
2
[ Dont merge ] verifying jax ai image
#257 opened Sep 30, 2025 by parambole • Draft
1
[Do not Merge] Test maxdiffusion workload on gpu runner
#255 opened Sep 26, 2025 by Rohan-Bierneni • Draft
1
Cross self attention switch
#251 opened Sep 19, 2025 by entrpn Loading…
2
[WAN] Use different sharding strategy for self and cross attention.
#250 opened Sep 18, 2025 by hyeygit Loading…
5
skip flash block sizes setting for cross attention.
#247 opened Sep 16, 2025 by entrpn Loading…
@coolkp
1
[DO Not Merge] Test Pin pytest version to 8.2.2
#225 opened Aug 8, 2025 by Rohan-Bierneni • Draft
2
[Do Not Merge] Testing deps in maxdiffusion image built using JAII
#215 opened Jul 29, 2025 by Rohan-Bierneni • Draft
1
Tf records from gcs
#213 opened Jul 28, 2025 by entrpn Loading…
1
Fix namedsharding
#190 opened Jun 25, 2025 by coolkp Loading…
Remove noop spmd_mode check, correct type annotations in attention flax
#170 opened May 1, 2025 by coolkp • Draft
Bump torch from 2.5.1 to 2.6.0 in the pip group across 1 directory dependencies Pull requests that update a dependency file python Pull requests that update python code
#165 opened Apr 22, 2025 by dependabot bot Loading…
Fix the GPU multi host segmentation fault on XPK
#129 opened Oct 25, 2024 by wang2yn84 Loading…
1
Replicated SDXL Quantization
#76 opened Jun 11, 2024 by mailvijayasingh Loading…
[Do not merge ]PTQ - AQT Integration
#72 opened Jun 4, 2024 by mailvijayasingh Loading…
Inference benchmarking metrics
#69 opened May 31, 2024 by mailvijayasingh Loading…
6
add back eval compilation cache
#26 opened Apr 22, 2024 by suexu1025 Loading…
[test][DO NOT MERGE] disable fori_loop
#11 opened Mar 21, 2024 by ManfeiBai • Draft
ProTip! Add no:assignee to see everything that’s not assigned.