{"payload":{"feedbackUrl":"https://github.com/orgs/community/discussions/53140","repo":{"id":233016299,"defaultBranch":"main","name":"flax","ownerLogin":"google","currentUserCanPush":false,"isFork":false,"isEmpty":false,"createdAt":"2020-01-10T09:48:37.000Z","ownerAvatar":"https://avatars.githubusercontent.com/u/1342004?v=4","public":true,"private":false,"isOrgOwned":true},"refInfo":{"name":"","listCacheKey":"v0:1727508228.0","currentOid":""},"activityList":{"items":[{"before":"fa210c0a616d42e59270be744e01bfde63bcde86","after":"7de8b3d80bc8a05dc9bc92933b0e1db4bfea5c64","ref":"refs/heads/test_679881642","pushedAt":"2024-09-28T07:45:01.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix false positive tracer leaks in flax library.\n\nWe'd like flax itself never to accidentally trigger the jax leak detector when enabled,\nbut over the years a few longer-lived temporaries were introduced to the flax codebase\nthat do trigger it.  We remedy those cases:\n\n- in flax.core remove a redundant use of nonlocal scopes in pack implementation.\n- for nn.jit have the hashing helper datastructure maintain a weakref to module.\n- explicitly exempt a context capture used in module summary tabulation routine.\n\nPiperOrigin-RevId: 679881642","shortMessageHtmlLink":"Fix false positive tracer leaks in flax library."}},{"before":"f65c2a53bc4f2895dffc055930feb218de423178","after":"fa210c0a616d42e59270be744e01bfde63bcde86","ref":"refs/heads/test_679881642","pushedAt":"2024-09-28T07:26:01.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix false positive tracer leaks in flax library.\n\nWe'd like flax itself never to accidentally trigger the jax leak detector when enabled,\nbut over the years a few longer-lived temporaries were introduced to the flax codebase\nthat do trigger it.  We remedy those cases:\n\n- in flax.core remove a redundant use of nonlocal scopes in pack implementation.\n- for nn.jit have the hashing helper datastructure maintain a weakref to module.\n- explicitly exempt a context capture used in module summary tabulation routine.\n\nPiperOrigin-RevId: 679881642","shortMessageHtmlLink":"Fix false positive tracer leaks in flax library."}},{"before":null,"after":"f65c2a53bc4f2895dffc055930feb218de423178","ref":"refs/heads/test_679881642","pushedAt":"2024-09-28T07:23:48.000Z","pushType":"branch_creation","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix false positive tracer leaks in flax library.\n\nWe'd like flax itself never to accidentally trigger the jax leak detector when enabled,\nbut over the years a few longer-lived temporaries were introduced to the flax codebase\nthat do trigger it.  We remedy those cases:\n\n- in flax.core remove a redundant use of nonlocal scopes in pack implementation.\n- for nn.jit have the hashing helper datastructure maintain a weakref to module.\n- explicitly exempt a context capture used in module summary tabulation routine.\n\nPiperOrigin-RevId: 679881642","shortMessageHtmlLink":"Fix false positive tracer leaks in flax library."}},{"before":"c0573372060966e6b322463f0ba5bfcb958ef5e7","after":null,"ref":"refs/heads/test_678109450","pushedAt":"2024-09-27T06:38:03.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"}},{"before":"37f643daf6d191bc6aa43c872adea9f14b279ee0","after":"c0573372060966e6b322463f0ba5bfcb958ef5e7","ref":"refs/heads/main","pushedAt":"2024-09-27T06:38:02.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix error when printing module with no fields.\n\n---\n\nCurrently, the edge case code will cause an error:\n\n```\nprint(nn.Module())\n```\n\nPiperOrigin-RevId: 679456418","shortMessageHtmlLink":"Fix error when printing module with no fields."}},{"before":"db129ccf98ccf56528f2037359603b9a11d245d8","after":"c0573372060966e6b322463f0ba5bfcb958ef5e7","ref":"refs/heads/test_678109450","pushedAt":"2024-09-27T06:38:01.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix error when printing module with no fields.\n\n---\n\nCurrently, the edge case code will cause an error:\n\n```\nprint(nn.Module())\n```\n\nPiperOrigin-RevId: 679456418","shortMessageHtmlLink":"Fix error when printing module with no fields."}},{"before":"e0b70ac64299cd6f147b7d789305590b50b51084","after":"db129ccf98ccf56528f2037359603b9a11d245d8","ref":"refs/heads/test_678109450","pushedAt":"2024-09-27T06:28:44.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Fix error when printing module with no fields.\n\n---\n\nCurrently, the edge case code will cause an error:\n\n```\nprint(nn.Module())\n```\n\nPiperOrigin-RevId: 678109450","shortMessageHtmlLink":"Fix error when printing module with no fields."}},{"before":"e7dba2823b8688e3c978299ee8f1701d1617962b","after":null,"ref":"refs/heads/nnx-remove-registry","pushedAt":"2024-09-26T18:37:21.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"}},{"before":"9f44f8114b28abe1ad8bd8f3191e1af5e4224a5b","after":"37f643daf6d191bc6aa43c872adea9f14b279ee0","ref":"refs/heads/main","pushedAt":"2024-09-26T18:37:19.000Z","pushType":"push","commitsCount":2,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #4205 from google:nnx-remove-registry\n\nPiperOrigin-RevId: 679230333","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2531091386\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/4205\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/4205/hovercard\" href=\"https://github.com/google/flax/pull/4205\">#4205</a> from google:nnx-remove-registry"}},{"before":"8b37d1abe3463e323520a865dd190c78c6569206","after":"9f44f8114b28abe1ad8bd8f3191e1af5e4224a5b","ref":"refs/heads/main","pushedAt":"2024-09-26T18:35:32.000Z","pushType":"push","commitsCount":2,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #4230 from IvyZX:pure-dict\n\nPiperOrigin-RevId: 679230175","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2549302797\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/4230\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/4230/hovercard\" href=\"https://github.com/google/flax/pull/4230\">#4230</a> from IvyZX:pure-dict"}},{"before":"26cd1f9f443b62c74e040163bf3a36b78a548a67","after":"580237c4e9cc60f8922047b7b82ef3533686d050","ref":"refs/heads/flaxlib","pushedAt":"2024-09-26T16:39:30.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] add flaxlib","shortMessageHtmlLink":"[nnx] add flaxlib"}},{"before":"650571ea36f0294a4e8b25bd4481eb522e4d0087","after":"26cd1f9f443b62c74e040163bf3a36b78a548a67","ref":"refs/heads/flaxlib","pushedAt":"2024-09-26T16:33:50.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] add flaxlib","shortMessageHtmlLink":"[nnx] add flaxlib"}},{"before":"4f62541a9748fb8cad2dae592beded28563bba37","after":"e7dba2823b8688e3c978299ee8f1701d1617962b","ref":"refs/heads/nnx-remove-registry","pushedAt":"2024-09-26T15:53:09.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"simplify graph.py","shortMessageHtmlLink":"simplify graph.py"}},{"before":"65d4193f3e655f4ca73828e7e1f45c461eace2a1","after":null,"ref":"refs/heads/nnx-randomness-guide","pushedAt":"2024-09-26T14:30:33.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"}},{"before":"fc19c5dcc136b17d7a2c406831126d93bbdac6d7","after":"8b37d1abe3463e323520a865dd190c78c6569206","ref":"refs/heads/main","pushedAt":"2024-09-26T14:30:31.000Z","pushType":"push","commitsCount":2,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #4216 from google:nnx-randomness-guide\n\nPiperOrigin-RevId: 679140602","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2541143781\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/4216\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/4216/hovercard\" href=\"https://github.com/google/flax/pull/4216\">#4216</a> from google:nnx-randomness-guide"}},{"before":"54fde939907139db9f8e81bd37858d84e9e69e90","after":"65d4193f3e655f4ca73828e7e1f45c461eace2a1","ref":"refs/heads/nnx-randomness-guide","pushedAt":"2024-09-26T14:16:58.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] add Randomness guide","shortMessageHtmlLink":"[nnx] add Randomness guide"}},{"before":"d9bd85c16cb454adf5c73169bbe1d13348b61078","after":"54fde939907139db9f8e81bd37858d84e9e69e90","ref":"refs/heads/nnx-randomness-guide","pushedAt":"2024-09-26T14:14:56.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] add Randomness guide","shortMessageHtmlLink":"[nnx] add Randomness guide"}},{"before":"fc19c5dcc136b17d7a2c406831126d93bbdac6d7","after":null,"ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T23:54:21.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"}},{"before":"9c86396c39b9484ca108fca49348b0b75a9c8316","after":"fc19c5dcc136b17d7a2c406831126d93bbdac6d7","ref":"refs/heads/main","pushedAt":"2024-09-24T23:54:19.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678446080","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"00a9393e9ede3dc266a69fb0a2b4c65846f17e9a","after":"fc19c5dcc136b17d7a2c406831126d93bbdac6d7","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T23:54:18.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678446080","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"c7a7142df2e0fce654f091eec4b0a35bfd5032aa","after":"00a9393e9ede3dc266a69fb0a2b4c65846f17e9a","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T23:48:13.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678351335","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"6d2355d9129ddc508a30a3a18487b2d6abb7b219","after":"c7a7142df2e0fce654f091eec4b0a35bfd5032aa","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T22:49:06.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678351335","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"e3772b2d01ba8aca7a09644ed66d4bc072c8782b","after":"9c86396c39b9484ca108fca49348b0b75a9c8316","ref":"refs/heads/main","pushedAt":"2024-09-24T21:46:28.000Z","pushType":"push","commitsCount":2,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #4220 from IvyZX:jit-guide\n\nPiperOrigin-RevId: 678404371","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2543911946\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/4220\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/4220/hovercard\" href=\"https://github.com/google/flax/pull/4220\">#4220</a> from IvyZX:jit-guide"}},{"before":"9cf4b3015a41ade4e2d9aadce2f53b722668d5a3","after":"6d2355d9129ddc508a30a3a18487b2d6abb7b219","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T20:08:00.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678351335","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"9644a878dc4d352161bc4c6c9148c3fa5b2fafb0","after":"9cf4b3015a41ade4e2d9aadce2f53b722668d5a3","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T20:06:13.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678351335","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":null,"after":"2d6aa629921bab06c1e233373a00247007c95c1f","ref":"refs/heads/nnx-train-state-toy-example","pushedAt":"2024-09-24T19:49:48.000Z","pushType":"branch_creation","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] revive TrainState toy example","shortMessageHtmlLink":"[nnx] revive TrainState toy example"}},{"before":null,"after":"9644a878dc4d352161bc4c6c9148c3fa5b2fafb0","ref":"refs/heads/test_678351335","pushedAt":"2024-09-24T19:29:12.000Z","pushType":"branch_creation","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 678351335","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"a7bcfb9f6c453fcf61015d98dfcfa89ae12df930","after":"815f9f531265b1d6d508c792bd356a33f8fac298","ref":"refs/heads/test_677843398","pushedAt":"2024-09-24T18:32:56.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 677843398","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"1a6c5fb044a3fe06b8821ec1b73fa0b45e8da739","after":"a7bcfb9f6c453fcf61015d98dfcfa89ae12df930","ref":"refs/heads/test_677843398","pushedAt":"2024-09-24T18:32:00.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars.githubusercontent.com/in/44061?s=80&v=4"},"commit":{"message":"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.\n\nPiperOrigin-RevId: 677843398","shortMessageHtmlLink":"Update libraries to use JAX's limited (and ill-advised) trace-state-q…"}},{"before":"de289a6de4cb9e26fd6968467cb1caaa406fa8d4","after":"d9bd85c16cb454adf5c73169bbe1d13348b61078","ref":"refs/heads/nnx-randomness-guide","pushedAt":"2024-09-24T10:49:10.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] add Randomness guide","shortMessageHtmlLink":"[nnx] add Randomness guide"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"startCursor":"Y3Vyc29yOnYyOpK7MjAyNC0wOS0yOFQwNzo0NTowMS4wMDAwMDBazwAAAATDQOa-","endCursor":"Y3Vyc29yOnYyOpK7MjAyNC0wOS0yNFQxMDo0OToxMC4wMDAwMDBazwAAAAS_Bi1e"}},"title":"Activity · google/flax"}