Skip to content

Actions: google-deepmind/dm-haiku

Actions

docs

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
docs.yml
364 workflow runs
364 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on. docs #2290: Pull request #796 synchronize by copybara-service bot
October 3, 2024 02:23 58s test_681582933
test_681582933
October 3, 2024 02:23 58s
Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on. docs #2289: Pull request #796 opened by copybara-service bot
October 3, 2024 01:26 48s test_681582933
test_681582933
October 3, 2024 01:26 48s
Update libraries to use JAX's limited (and ill-advised) trace-state-q… docs #2288: Commit af844e9 pushed by copybara-service bot
September 25, 2024 08:37 48s main
main
September 25, 2024 08:37 48s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2287: Pull request #795 synchronize by copybara-service bot
September 25, 2024 08:19 1m 0s test_678332661
test_678332661
September 25, 2024 08:19 1m 0s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2286: Pull request #795 opened by copybara-service bot
September 24, 2024 19:29 54s test_678332661
test_678332661
September 24, 2024 19:29 54s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2285: Pull request #794 synchronize by copybara-service bot
September 24, 2024 18:33 55s test_677843398
test_677843398
September 24, 2024 18:33 55s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2284: Pull request #794 synchronize by copybara-service bot
September 24, 2024 18:31 54s test_677843398
test_677843398
September 24, 2024 18:31 54s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2283: Pull request #794 synchronize by copybara-service bot
September 24, 2024 02:13 1m 0s test_677843398
test_677843398
September 24, 2024 02:13 1m 0s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2282: Pull request #794 synchronize by copybara-service bot
September 24, 2024 00:27 1m 20s test_677843398
test_677843398
September 24, 2024 00:27 1m 20s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2281: Pull request #794 synchronize by copybara-service bot
September 24, 2024 00:15 54s test_677843398
test_677843398
September 24, 2024 00:15 54s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2280: Pull request #794 synchronize by copybara-service bot
September 23, 2024 23:24 50s test_677843398
test_677843398
September 23, 2024 23:24 50s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2279: Pull request #794 synchronize by copybara-service bot
September 23, 2024 20:27 54s test_677843398
test_677843398
September 23, 2024 20:27 54s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2278: Pull request #794 synchronize by copybara-service bot
September 23, 2024 19:53 52s test_677843398
test_677843398
September 23, 2024 19:53 52s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2277: Pull request #794 synchronize by copybara-service bot
September 23, 2024 19:52 1m 6s test_677843398
test_677843398
September 23, 2024 19:52 1m 6s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2276: Pull request #794 synchronize by copybara-service bot
September 23, 2024 19:34 49s test_677843398
test_677843398
September 23, 2024 19:34 49s
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. docs #2275: Pull request #794 opened by copybara-service bot
September 23, 2024 17:58 52s test_677843398
test_677843398
September 23, 2024 17:58 52s
Correct an oversight in TruncatedNormal's documentation. docs #2274: Pull request #793 synchronize by copybara-service bot
September 11, 2024 14:51 50s test_672980025
test_672980025
September 11, 2024 14:51 50s
Correct an oversight in TruncatedNormal's documentation. docs #2273: Commit 4773949 pushed by copybara-service bot
September 11, 2024 14:51 52s main
main
September 11, 2024 14:51 52s
Correct an oversight in TruncatedNormal's documentation. docs #2272: Pull request #793 synchronize by copybara-service bot
September 11, 2024 14:46 58s test_672980025
test_672980025
September 11, 2024 14:46 58s
Correct an oversight in TruncatedNormal's documentation. docs #2271: Pull request #793 opened by copybara-service bot
September 11, 2024 13:56 1m 1s test_672980025
test_672980025
September 11, 2024 13:56 1m 1s
[Haiku] Add shard_map flops estimate to the jaxpr_info. docs #2270: Pull request #791 synchronize by copybara-service bot
August 30, 2024 13:28 54s test_669298816
test_669298816
August 30, 2024 13:28 54s
[Haiku] Add shard_map flops estimate to the jaxpr_info. docs #2269: Commit 399ec5b pushed by copybara-service bot
August 30, 2024 13:28 48s main
main
August 30, 2024 13:28 48s
[Haiku] Add shard_map flops estimate to the jaxpr_info. docs #2268: Pull request #791 synchronize by copybara-service bot
August 30, 2024 13:23 1m 13s test_669298816
test_669298816
August 30, 2024 13:23 1m 13s
[Haiku] Add shard_map flops estimate to the jaxpr_info. docs #2267: Pull request #791 opened by copybara-service bot
August 30, 2024 13:03 59s test_669298816
test_669298816
August 30, 2024 13:03 59s
Remove uses of deprecated xb, xc, and xe abbreviation from jax.interpreters.xla docs #2266: Pull request #789 synchronize by copybara-service bot
August 2, 2024 07:00 56s test_658556158
test_658556158
August 2, 2024 07:00 56s