{"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:1714785326.0","currentOid":""},"activityList":{"items":[{"before":null,"after":"96116112d52dba22627a8de73c7bf3efd7fe1148","ref":"refs/heads/test_630547382","pushedAt":"2024-05-04T01:15:26.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":"Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical specification\n\nPiperOrigin-RevId: 630547382","shortMessageHtmlLink":"Add support for <code>jax.sharding.PartitionSpec.UNCONSTRAINED</code> in logical…"}},{"before":"47ad8982afc9700079369d4ffac08d686fd44b55","after":"0459456f491f906f86eb074e4123d759832e0eb7","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T18:12:00.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"859435e4225706d25eba2138d51fb101e6e70f4c","after":"40c7eaa74817a1716d3f64ad70a292e975837753","ref":"refs/heads/main","pushedAt":"2024-05-03T17:42:51.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 #3901 from NeilGirdhar:fix_threading\n\nPiperOrigin-RevId: 630432840","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2277606164\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/3901\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/3901/hovercard\" href=\"https://github.com/google/flax/pull/3901\">#3901</a> from NeilGirdhar:fix_threading"}},{"before":"bce897b0a4bd6d5bb00d6c66e8b6385a3fea7c97","after":"47ad8982afc9700079369d4ffac08d686fd44b55","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T16:30:06.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"bed68d0d0d091266f88f26a92e9bd4f18753613c","after":"bce897b0a4bd6d5bb00d6c66e8b6385a3fea7c97","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T16:01:51.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"7b74eea2b652427b0bb827e633665f49144c1dcc","after":"bed68d0d0d091266f88f26a92e9bd4f18753613c","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T14:58:57.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"d36bb2db4beda452411e86b79a15e624fd55c5f0","after":"7b74eea2b652427b0bb827e633665f49144c1dcc","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T13:48:06.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":null,"after":"e70f6bbe11571f6a17869d83c11d59fe41e7ac69","ref":"refs/heads/linen-test-jit-state","pushedAt":"2024-05-03T11:33:36.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":"[linen] test jit caching with state updates","shortMessageHtmlLink":"[linen] test jit caching with state updates"}},{"before":"57a0dd5504e0c910f8b6723a70ae17bb3a737aab","after":"d36bb2db4beda452411e86b79a15e624fd55c5f0","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-03T10:34:13.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"859435e4225706d25eba2138d51fb101e6e70f4c","after":null,"ref":"refs/heads/test_630155744","pushedAt":"2024-05-02T20:20:05.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":"d0e080da1861bdddec1b2e10e5dab094f505b933","after":"859435e4225706d25eba2138d51fb101e6e70f4c","ref":"refs/heads/main","pushedAt":"2024-05-02T20:20:04.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":"Copybara import of the project:\n\n--\nd1935c71686533b2aa3d395d47395bf21457bffa by Neil Girdhar <mistersheik@gmail.com>:\n\nMake FlatState a Mapping instead of a dict\n\nAlso, make traverse_util.flatten_dict accept mappings.\n\nFixes https://github.com/google/flax/issues/3879\n\nPiperOrigin-RevId: 630158841","shortMessageHtmlLink":"Copybara import of the project:"}},{"before":"d7f5c5007e69af1e624c1e6fc280cf7dc08635e4","after":"859435e4225706d25eba2138d51fb101e6e70f4c","ref":"refs/heads/test_630155744","pushedAt":"2024-05-02T20:20:03.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":"Copybara import of the project:\n\n--\nd1935c71686533b2aa3d395d47395bf21457bffa by Neil Girdhar <mistersheik@gmail.com>:\n\nMake FlatState a Mapping instead of a dict\n\nAlso, make traverse_util.flatten_dict accept mappings.\n\nFixes https://github.com/google/flax/issues/3879\n\nPiperOrigin-RevId: 630158841","shortMessageHtmlLink":"Copybara import of the project:"}},{"before":null,"after":"d7f5c5007e69af1e624c1e6fc280cf7dc08635e4","ref":"refs/heads/test_630155744","pushedAt":"2024-05-02T20:14:14.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":"Copybara import of the project:\n\n--\nd1935c71686533b2aa3d395d47395bf21457bffa by Neil Girdhar <mistersheik@gmail.com>:\n\nMake FlatState a Mapping instead of a dict\n\nAlso, make traverse_util.flatten_dict accept mappings.\n\nFixes https://github.com/google/flax/issues/3879\n\nPiperOrigin-RevId: 630155744","shortMessageHtmlLink":"Copybara import of the project:"}},{"before":"aba4d50fcc593043882c2d3f8e1bbae17008418a","after":"57a0dd5504e0c910f8b6723a70ae17bb3a737aab","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T15:45:55.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"d4d4478e6807413bcd1b6412c34c2ccdb4d9b9e7","after":"aba4d50fcc593043882c2d3f8e1bbae17008418a","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T14:25:46.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"a6d781d41d467a52bb11845f342d20c0dce7fe61","after":"d4d4478e6807413bcd1b6412c34c2ccdb4d9b9e7","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T14:01:26.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"77dc0fb1bbd09e9e0f1ba988b010b265a5c055e4","after":"a6d781d41d467a52bb11845f342d20c0dce7fe61","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T12:34:25.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"61806b5e396eebac048a35374c472d6c4a12ca82","after":"77dc0fb1bbd09e9e0f1ba988b010b265a5c055e4","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T11:44:14.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"15def59a80f2ea6f3d2b437af86c0b8628a819a8","after":"61806b5e396eebac048a35374c472d6c4a12ca82","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T11:01:05.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"ae5d66d3ad7b54d7bd9544baade1551d4412044a","after":"d0e080da1861bdddec1b2e10e5dab094f505b933","ref":"refs/heads/main","pushedAt":"2024-05-02T10:52:42.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 #3880 from NeilGirdhar:fix_mappings\n\nPiperOrigin-RevId: 630013257","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2258040346\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/3880\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/3880/hovercard\" href=\"https://github.com/google/flax/pull/3880\">#3880</a> from NeilGirdhar:fix_mappings"}},{"before":"8f02346a9284a25e96c2f0dfc3ca79beea9c4a96","after":"15def59a80f2ea6f3d2b437af86c0b8628a819a8","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T10:33:21.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"8813ec9cb2594d958a426a531590fe9e21aabe56","after":"8f02346a9284a25e96c2f0dfc3ca79beea9c4a96","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T10:18:08.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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"568887f2544a018fcfd16668dda3a8d9dba9468e","after":"8813ec9cb2594d958a426a531590fe9e21aabe56","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T09:31: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] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"895b78337169bfdc2f0fc6128adb8518aa6ace3e","after":"568887f2544a018fcfd16668dda3a8d9dba9468e","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-02T09:30:35.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"cgarciae","name":"Cristian Garcia","path":"/cgarciae","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/5862228?s=80&v=4"},"commit":{"message":"[nnx] enable pytype on CI","shortMessageHtmlLink":"[nnx] enable pytype on CI"}},{"before":"af45534f19d2424985c0e690d995375433be8220","after":"7ee6ba635b57c59554e63b2ee603cb079faa2e89","ref":"refs/heads/test_629615743","pushedAt":"2024-05-01T18:35:55.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":"Add a get_inputs_kernel_bias method to the DenseGeneral layer.\n\nThis method returns the inputs, kernel, bias, dot_dimension_nums, and out_shape_sequence_int. This is useful for users who want to use the DenseGeneral layer with custom kernel containing the dot_general step.\n\nPiperOrigin-RevId: 629615743","shortMessageHtmlLink":"Add a get_inputs_kernel_bias method to the DenseGeneral layer."}},{"before":"2026c145a6cad90fadfe477f5399eda69a1042e8","after":"af45534f19d2424985c0e690d995375433be8220","ref":"refs/heads/test_629615743","pushedAt":"2024-05-01T14:45:29.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":"Add a get_inputs_kernel_bias method to the DenseGeneral layer.\n\nThis method returns the inputs, kernel, bias, dot_dimension_nums, and out_shape_sequence_int. This is useful for users who want to use the DenseGeneral layer with custom kernel containing the dot_general step.\n\nPiperOrigin-RevId: 629615743","shortMessageHtmlLink":"Add a get_inputs_kernel_bias method to the DenseGeneral layer."}},{"before":"5077e566cd0bd1e85295445ff41a7d7be09636f3","after":"895b78337169bfdc2f0fc6128adb8518aa6ace3e","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-01T12:16: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] enable pytype and mypy on CI","shortMessageHtmlLink":"[nnx] enable pytype and mypy on CI"}},{"before":"263767c5f7d970837f199b3e3cdb34fb21382e37","after":"5077e566cd0bd1e85295445ff41a7d7be09636f3","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-01T11:39:51.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] enable pytype and mypy on CI","shortMessageHtmlLink":"[nnx] enable pytype and mypy on CI"}},{"before":"43434adc278e1a3a27906a25de68e5352861f2b1","after":"263767c5f7d970837f199b3e3cdb34fb21382e37","ref":"refs/heads/nnx-enable-mypy-pytype-ci","pushedAt":"2024-05-01T07:31:24.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] enable pytype and mypy on CI","shortMessageHtmlLink":"[nnx] enable pytype and mypy on CI"}},{"before":"2c7d7cd9e2c393b3c4ff44ef0454beea6c9de28f","after":"ae5d66d3ad7b54d7bd9544baade1551d4412044a","ref":"refs/heads/main","pushedAt":"2024-05-01T07:13:09.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 #3895 from chiamp:codecov\n\nPiperOrigin-RevId: 629645344","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2272786586\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/flax/issues/3895\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/flax/pull/3895/hovercard\" href=\"https://github.com/google/flax/pull/3895\">#3895</a> from chiamp:codecov"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"cursor":"djE6ks8AAAAEQTbM4QA","startCursor":null,"endCursor":null}},"title":"Activity · google/flax"}