Skip to content

Actions: google/flax

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
7,200 workflow runs
7,200 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Add intermediate value captures (extends #4925) Google GitHub Admin: Actions Workflow Security Scan #535: Pull request #5257 synchronize by samanklesaria
51s samanklesaria:capture_intms
samanklesaria:capture_intms
51s
Use jax.sharding.get_abstract_mesh() in nnx/transforms. Flax - Test #4454: Pull request #5315 opened by copybara-service bot
5m 4s test_879615724
test_879615724
5m 4s
Use jax.sharding.get_abstract_mesh() in nnx/transforms. Google GitHub Admin: Actions Workflow Security Scan #534: Pull request #5315 opened by copybara-service bot
42s test_879615724
test_879615724
42s
PR #5315 CodeQL #4795: by copybara-service bot
1m 31s refs/pull/5315/head
refs/pull/5315/head
1m 31s
Use jax.sharding.get_abstract_mesh() in nnx/transforms. Flax - Test #4453: Commit 340b5fa pushed by copybara-service bot
4m 49s test_879615724
test_879615724
4m 49s
CI - with JAX nightly CI - with JAX nightly #432: Scheduled
5m 56s main
main
5m 56s
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
ProTip! You can narrow down the results and go further in time using created:<2026-03-05 or the other filters available.