-
Notifications
You must be signed in to change notification settings - Fork 56
Pull requests: NVIDIA/JAX-Toolbox
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Update ml_dtypes to the latest version after pip-finalize.
#1250
by DwarKapex
was merged Jan 15, 2025
Loading…
Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs
#1249
by olupton
was merged Jan 15, 2025
Loading…
[nsys-jax] Add ratio of hidden communication time to total communication time
#1241
by sfvaroglu
was merged Jan 16, 2025
Loading…
Remove deprecated flag xla_gpu_enable_triton_softmax_fusion.
#1240
by sergachev
was merged Jan 21, 2025
Loading…
Remove deprecated flag xla_gpu_enable_triton_softmax_fusion.
#1239
by sergachev
was closed Jan 10, 2025
Loading…
CI: run MaxText tests on AWS with NGC release candidate images
#1237
by olupton
was merged Jan 16, 2025
Loading…
Previous Next
ProTip!
What’s not been updated in a month: updated:<2024-12-22.