Run Flax dependency tests

remove call to `F.pad`, improved calculation of `memory_count` #13031

Sign in to view logs

Re-run triggered January 21, 2025 22:00

hlky

#10620

bm-synth:inplace_sum_and_remove_padding_and_better_memory_count

Status Success

Total duration 38s

Artifacts –

pr_flax_dependency_test.yml

on: pull_request

check_flax_dependencies