Run Flax dependency tests

[feat] implement `record_stream` when using CUDA streams during group offloading #14229

Sign in to view logs

Triggered via pull request March 17, 2025 06:53

sayakpaul

opened #11081

record-streams

Status Success

Total duration 35s

Artifacts –

pr_flax_dependency_test.yml

on: pull_request

check_flax_dependencies