Skip to content

Actions: google/flax

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
6,616 workflow runs
6,616 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Merge pull request #5247 from samanklesaria:eval_shape_sharding Flax - Test #4452: Commit 49e34d1 pushed by copybara-service bot
11m 38s main
main
11m 38s
Push on main CodeQL #4794: by copybara-service bot
1m 33s main
main
1m 33s
Add 'opt_sharding' option for Variables Google GitHub Admin: Actions Workflow Security Scan #533: Pull request #5231 synchronize by samanklesaria
1m 40s samanklesaria:opt_sharding
samanklesaria:opt_sharding
1m 40s
Add 'opt_sharding' option for Variables Flax - Test #4451: Pull request #5231 synchronize by samanklesaria
4m 28s samanklesaria:opt_sharding
samanklesaria:opt_sharding
4m 28s
feat(nnx): add GQA support to MultiHeadAttention Google GitHub Admin: Actions Workflow Security Scan #532: Pull request #5259 synchronize by samanklesaria
48s ayulockedin:feat/gqa-support-mha
ayulockedin:feat/gqa-support-mha
48s
feat(nnx): add GQA support to MultiHeadAttention Flax - Test #4450: Pull request #5259 synchronize by samanklesaria
12m 56s ayulockedin:feat/gqa-support-mha
ayulockedin:feat/gqa-support-mha
12m 56s
Add 'opt_sharding' option for Variables Google GitHub Admin: Actions Workflow Security Scan #531: Pull request #5231 synchronize by samanklesaria
42s samanklesaria:opt_sharding
samanklesaria:opt_sharding
42s
Add 'opt_sharding' option for Variables Flax - Test #4449: Pull request #5231 synchronize by samanklesaria
13m 51s samanklesaria:opt_sharding
samanklesaria:opt_sharding
13m 51s
Add 'opt_sharding' option for Variables Google GitHub Admin: Actions Workflow Security Scan #530: Pull request #5231 synchronize by samanklesaria
44s samanklesaria:opt_sharding
samanklesaria:opt_sharding
44s
Add 'opt_sharding' option for Variables Flax - Test #4448: Pull request #5231 synchronize by samanklesaria
36s samanklesaria:opt_sharding
samanklesaria:opt_sharding
36s
Generalize out_sharding to work with NamedSharding and Format Google GitHub Admin: Actions Workflow Security Scan #529: Pull request #5246 synchronize by samanklesaria
46s samanklesaria:generalize_out_sharding
samanklesaria:generalize_out_sharding
46s
Generalize out_sharding to work with NamedSharding and Format Flax - Test #4447: Pull request #5246 synchronize by samanklesaria
11m 27s samanklesaria:generalize_out_sharding
samanklesaria:generalize_out_sharding
11m 27s
Add sharding propagation support in nnx.eval_shape (clone of #5111) Flax - Test #4446: Pull request #5247 synchronize by samanklesaria
13m 12s samanklesaria:eval_shape_sharding
samanklesaria:eval_shape_sharding
13m 12s
Add sharding propagation support in nnx.eval_shape (clone of #5111) Google GitHub Admin: Actions Workflow Security Scan #528: Pull request #5247 synchronize by samanklesaria
48s samanklesaria:eval_shape_sharding
samanklesaria:eval_shape_sharding
48s
CI - with JAX nightly CI - with JAX nightly #431: Scheduled
6m 9s main
main
6m 9s
avoid __eq__ in test_iter_graph Flax - Test #4445: Commit 39b7613 pushed by copybara-service bot
11m 34s main
main
11m 34s
PR #5313 CodeQL #4793: by copybara-service bot
1m 10s refs/pull/5313/head
refs/pull/5313/head
1m 10s
avoid __eq__ in test_iter_graph Flax - Test #4444: Commit 39b7613 pushed by copybara-service bot
33s test_878762402
test_878762402
33s
Push on main CodeQL #4792: by copybara-service bot
1m 44s main
main
1m 44s
add nnx_basics_tree.ipynb Google GitHub Admin: Actions Workflow Security Scan #527: Pull request #5293 synchronize by cgarciae
43s nnx-basics-tree
nnx-basics-tree
43s
add nnx_basics_tree.ipynb Flax - Test #4443: Pull request #5293 synchronize by cgarciae
11m 29s nnx-basics-tree
nnx-basics-tree
11m 29s
PR #5293 CodeQL #4791: by cgarciae
1m 12s refs/pull/5293/head
refs/pull/5293/head
1m 12s
avoid __eq__ in test_iter_graph Google GitHub Admin: Actions Workflow Security Scan #526: Pull request #5313 synchronize by copybara-service bot
44s test_878762402
test_878762402
44s
avoid __eq__ in test_iter_graph Flax - Test #4442: Pull request #5313 synchronize by copybara-service bot
12m 25s test_878762402
test_878762402
12m 25s
avoid __eq__ in test_iter_graph Flax - Test #4441: Commit b8d26c7 pushed by copybara-service bot
11m 59s test_878762402
test_878762402
11m 59s