Skip to content

Actions: google/flax

Actions

Flax - Test

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
flax_test.yml
3,022 workflow runs
3,022 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[pmap] Update jax_utils.unreplicate for performant use when jax.config.jax_pmap_shmap_merge is True. Flax - Test #3618: Pull request #5153 opened by copybara-service bot
5m 41s test_845027363
test_845027363
5m 41s
[pmap] Update jax_utils.unreplicate for performant use when `jax.co… Flax - Test #3617: Commit 9f725d4 pushed by copybara-service bot
4m 42s test_845027363
test_845027363
4m 42s
[pmap] Avoid degraded performance under the new jax.pmap. Flax - Test #3616: Pull request #5152 opened by copybara-service bot
5m 50s test_846723995
test_846723995
5m 50s
[pmap] Avoid degraded performance under the new jax.pmap. Flax - Test #3615: Commit 3f5337c pushed by copybara-service bot
5m 59s test_846723995
test_846723995
5m 59s
Merge pull request #5149 from IvyZX:release Flax - Test #3614: Commit d179184 pushed by copybara-service bot
21m 47s main
main
21m 47s
New release 0.12.2 Flax - Test #3613: Pull request #5149 synchronize by IvyZX
18m 6s IvyZX:release
IvyZX:release
18m 6s
New release 0.12.2 Flax - Test #3612: Pull request #5149 synchronize by IvyZX
10m 39s IvyZX:release
IvyZX:release
10m 39s
New release 0.12.2 Flax - Test #3611: Pull request #5149 synchronize by IvyZX
37s IvyZX:release
IvyZX:release
37s
Flax 0.12.2 release Flax - Test #3610: Pull request #5150 synchronize by copybara-service bot
3m 21s test_846381464
test_846381464
3m 21s
Flax 0.12.2 release Flax - Test #3609: Commit 1e5f854 pushed by copybara-service bot
3m 22s test_846381464
test_846381464
3m 22s
Flax 0.12.2 release Flax - Test #3608: Pull request #5150 synchronize by copybara-service bot
3m 49s test_846381464
test_846381464
3m 49s
Flax 0.12.2 release Flax - Test #3607: Commit d3de51b pushed by copybara-service bot
3m 46s test_846381464
test_846381464
3m 46s
Flax 0.12.2 release Flax - Test #3606: Pull request #5150 synchronize by copybara-service bot
2m 32s test_846381464
test_846381464
2m 32s
Flax 0.12.2 release Flax - Test #3605: Commit 8a24acb pushed by copybara-service bot
2m 31s test_846381464
test_846381464
2m 31s
Merge pull request #5144 from google:fix-ci-5143 Flax - Test #3604: Commit 9213cfd pushed by copybara-service bot
4m 30s main
main
4m 30s
Flax 0.12.2 release Flax - Test #3603: Pull request #5150 opened by copybara-service bot
1m 49s test_846381464
test_846381464
1m 49s
Flax 0.12.2 release Flax - Test #3602: Commit 448e63a pushed by copybara-service bot
2m 4s test_846381464
test_846381464
2m 4s
New release 0.12.2 Flax - Test #3601: Pull request #5149 opened by IvyZX
3m 20s IvyZX:release
IvyZX:release
3m 20s
Have nnx.pop remove attributes from pytree__nodes Flax - Test #3600: Pull request #5147 opened by samanklesaria
3m 21s samanklesaria:issues/5146
samanklesaria:issues/5146
3m 21s
Fixed typos in the docstrings using antigravity Flax - Test #3599: Pull request #5145 opened by vfdev-5
7m 14s fix-more-typos
fix-more-typos
7m 14s
Temporary fix for failing CI Flax - Test #3598: Pull request #5144 synchronize by vfdev-5
16m 52s fix-ci-5143
fix-ci-5143
16m 52s
Temporary fix for failing CI Flax - Test #3597: Pull request #5144 opened by vfdev-5
49s fix-ci-5143
fix-ci-5143
49s
Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put Flax - Test #3596: Pull request #5101 synchronize by vfdev-5
14m 46s rewrite-prefetch_to_device
rewrite-prefetch_to_device
14m 46s
Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put Flax - Test #3595: Pull request #5101 synchronize by vfdev-5
3m 24s rewrite-prefetch_to_device
rewrite-prefetch_to_device
3m 24s
Typo fix in _cached_partial method Flax - Test #3594: Pull request #5142 opened by vfdev-5
4m 35s docs-typo-fix-_cached_partial
docs-typo-fix-_cached_partial
4m 35s