Actions: google-deepmind/torax
Actions
Showing runs from all workflows
2,500+ workflow runs
2,500+ workflow runs
jax_utils.while_loop_bounded with implementation='while_loop' be differentiable through closures.
Unittests
#20907:
Pull request #2303
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded with implementation='while_loop' be differentiable through closures.
Google GitHub Admin: Actions Workflow Security Scan
#1972:
Pull request #2303
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded with implementation='while_loop' be differentiable through closures.
GitHub Actions Scan
#453:
Pull request #2303
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded with `…
Unittests
#20906:
Commit 1de0f48
pushed
by
copybara-service
Bot
jax_utils.while_loop_bounded using the new JAX hijax API, which allows both custom VJP and custom JVP.
GitHub Actions Scan
#452:
Pull request #2302
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded using the new JAX hijax API, which allows both custom VJP and custom JVP.
Unittests
#20904:
Pull request #2302
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded using the new JAX hijax API, which allows both custom VJP and custom JVP.
Google GitHub Admin: Actions Workflow Security Scan
#1971:
Pull request #2302
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded using the new JAX h…
Unittests
#20903:
Commit fe3ce72
pushed
by
copybara-service
Bot
jax_utils.while_loop_bounded against jax.lax.while_loop`.
Google GitHub Admin: Actions Workflow Security Scan
#1969:
Pull request #2301
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded against jax.lax.while_loop`.
GitHub Actions Scan
#450:
Pull request #2301
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded against jax.lax.while_loop`.
Unittests
#20900:
Pull request #2301
opened
by
copybara-service
Bot
jax_utils.while_loop_bounded against jax.lax.while…
Unittests
#20899:
Commit aa897bd
pushed
by
copybara-service
Bot