{"payload":{"feedbackUrl":"https://github.com/orgs/community/discussions/53140","repo":{"id":154739597,"defaultBranch":"main","name":"jax","ownerLogin":"google","currentUserCanPush":false,"isFork":false,"isEmpty":false,"createdAt":"2018-10-25T21:25:02.000Z","ownerAvatar":"https://avatars.githubusercontent.com/u/1342004?v=4","public":true,"private":false,"isOrgOwned":true},"refInfo":{"name":"","listCacheKey":"v0:1718341794.0","currentOid":""},"activityList":{"items":[{"before":"6c79f392d2835b177d28e24cbdbfcb992e0318b2","after":"cbfa4d0c9e41f73fd217e05ba7f9c325ed9a1b16","ref":"refs/heads/test_643201055","pushedAt":"2024-06-14T05:23:48.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 `lowering_platforms` to `traced.lower()`\n\nPiperOrigin-RevId: 643201055","shortMessageHtmlLink":"Add lowering_platforms to traced.lower()"}},{"before":"30c064396ec128744655ecdaa279ac4c097e19bb","after":"512e614f2eb3188d2c64ef5580c418690cbf711e","ref":"refs/heads/test_643183904","pushedAt":"2024-06-14T05:23:35.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":"[Pallas] Make num_programs return an int if the grid is not dynamic\n\nPiperOrigin-RevId: 643183904","shortMessageHtmlLink":"[Pallas] Make num_programs return an int if the grid is not dynamic"}},{"before":"bc998db918431431c499fc16471a580329d6084e","after":"30c064396ec128744655ecdaa279ac4c097e19bb","ref":"refs/heads/test_643183904","pushedAt":"2024-06-14T05:17:16.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":"[Pallas] Make num_programs return an int if the grid is not dynamic\n\nPiperOrigin-RevId: 643183904","shortMessageHtmlLink":"[Pallas] Make num_programs return an int if the grid is not dynamic"}},{"before":null,"after":"6c79f392d2835b177d28e24cbdbfcb992e0318b2","ref":"refs/heads/test_643201055","pushedAt":"2024-06-14T05:09:54.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 `lowering_platforms` to `traced.lower()`\n\nPiperOrigin-RevId: 643201055","shortMessageHtmlLink":"Add lowering_platforms to traced.lower()"}},{"before":null,"after":"a7e9690afee5921ea777fc31adb32431b3cd7624","ref":"refs/heads/test_643147162","pushedAt":"2024-06-14T04:04:10.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":"[XLA:Mosaic] Add internal scratch VMEM\n\n- Make internal scratch size to configurable.\n- Pass the number of max sublanes allowed in scratch to apply vector layout pass.\n- Create a helper function to fetch internal scratch VMEM address.\n\nPiperOrigin-RevId: 643147162","shortMessageHtmlLink":"[XLA:Mosaic] Add internal scratch VMEM"}},{"before":"e12656002f424175b2903727a7c6529077d0e217","after":null,"ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T03:24:48.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":"a92fa547a0111f189261cfa67ecddba3124812ca","after":"e12656002f424175b2903727a7c6529077d0e217","ref":"refs/heads/main","pushedAt":"2024-06-14T03:24:47.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 643209848","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"e985bbf19c3a46e2f7a7ff5a0913798a94260b63","after":"e12656002f424175b2903727a7c6529077d0e217","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T03:24:45.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 643209848","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"b542e5837d8f775cb8a847f5c92bd2a84333d3b1","after":"e985bbf19c3a46e2f7a7ff5a0913798a94260b63","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T03:11: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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 642798417","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"a92fa547a0111f189261cfa67ecddba3124812ca","after":null,"ref":"refs/heads/test_643133791","pushedAt":"2024-06-14T02:53:58.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":"06ec7d1ad59968e742429a8480e18d56327796d0","after":"a92fa547a0111f189261cfa67ecddba3124812ca","ref":"refs/heads/main","pushedAt":"2024-06-14T02:53:56.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":"Re-land https://github.com/google/jax/pull/21847\n\nReverts 0bcc81ceb33e3065110e3dd56ca215dbb62f0a7b\n\nPiperOrigin-RevId: 643202512","shortMessageHtmlLink":"Re-land #21847"}},{"before":"ed2dd881cc599331a5f38efffa9abbfaae9af2c1","after":"a92fa547a0111f189261cfa67ecddba3124812ca","ref":"refs/heads/test_643133791","pushedAt":"2024-06-14T02:53: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":"Re-land https://github.com/google/jax/pull/21847\n\nReverts 0bcc81ceb33e3065110e3dd56ca215dbb62f0a7b\n\nPiperOrigin-RevId: 643202512","shortMessageHtmlLink":"Re-land #21847"}},{"before":"17d3dbe20f2e13124ee3fcebd936d53cef541dd4","after":"6fcf1f644344ad445ea9de69244c76d7d391e745","ref":"refs/heads/test_643078740","pushedAt":"2024-06-14T02:39:50.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":"Deprecate axis_env and tuple_args argument of xla_computation.\n\nPiperOrigin-RevId: 643078740","shortMessageHtmlLink":"Deprecate axis_env and tuple_args argument of xla_computation."}},{"before":"ce45e03c960ddaf9677b9f85ef4e3de39874969b","after":"ed2dd881cc599331a5f38efffa9abbfaae9af2c1","ref":"refs/heads/test_643133791","pushedAt":"2024-06-14T02:38:19.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":"Re-land https://github.com/google/jax/pull/21847\n\nReverts 0bcc81ceb33e3065110e3dd56ca215dbb62f0a7b\n\nPiperOrigin-RevId: 643133791","shortMessageHtmlLink":"Re-land #21847"}},{"before":"3a09000a2d8ada8ee7b4b8449c75b2d53b0a3823","after":"b542e5837d8f775cb8a847f5c92bd2a84333d3b1","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T02:28:52.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 642798417","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"d36878cd95275801058c5263e386c99b4913b4dd","after":"bc998db918431431c499fc16471a580329d6084e","ref":"refs/heads/test_643183904","pushedAt":"2024-06-14T01:45:07.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":"[Pallas] Make num_programs return an int if the grid is not dynamic\n\nPiperOrigin-RevId: 643183904","shortMessageHtmlLink":"[Pallas] Make num_programs return an int if the grid is not dynamic"}},{"before":null,"after":"d36878cd95275801058c5263e386c99b4913b4dd","ref":"refs/heads/test_643183904","pushedAt":"2024-06-14T01:34:32.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":"[Pallas] Make num_programs return an int if the grid is not dynamic\n\nPiperOrigin-RevId: 643183904","shortMessageHtmlLink":"[Pallas] Make num_programs return an int if the grid is not dynamic"}},{"before":"c030fb544f0c65ef162bd476986cae85e4dc510a","after":"3a09000a2d8ada8ee7b4b8449c75b2d53b0a3823","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T00:24:15.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 642798417","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"cb419c9de7b5813ccab5c62a63a7175db61c11d2","after":"c030fb544f0c65ef162bd476986cae85e4dc510a","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T00:22:43.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 642798417","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"792e1dfd46eea7442fca364f9933934952f1d6be","after":"cb419c9de7b5813ccab5c62a63a7175db61c11d2","ref":"refs/heads/test_642798417","pushedAt":"2024-06-14T00:21:30.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":"[Pallas] Don't actually vmap if we're vmapping over axis size 1\n\nPiperOrigin-RevId: 642798417","shortMessageHtmlLink":"[Pallas] Don't actually vmap if we're vmapping over axis size 1"}},{"before":"06ec7d1ad59968e742429a8480e18d56327796d0","after":null,"ref":"refs/heads/test_643145493","pushedAt":"2024-06-14T00:02:08.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":"b907242f3ea1eb3da99410fcf0e33032560e3209","after":"06ec7d1ad59968e742429a8480e18d56327796d0","ref":"refs/heads/main","pushedAt":"2024-06-14T00:02:06.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":"Reduce the matrix size in `testPmapMapVmapCombinations` to reduce the test run time.\n\nPiperOrigin-RevId: 643166085","shortMessageHtmlLink":"Reduce the matrix size in testPmapMapVmapCombinations to reduce the…"}},{"before":"bd73b98fc7c9faac94311f6f50d614cd3f95a33e","after":"06ec7d1ad59968e742429a8480e18d56327796d0","ref":"refs/heads/test_643145493","pushedAt":"2024-06-14T00:02:05.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":"Reduce the matrix size in `testPmapMapVmapCombinations` to reduce the test run time.\n\nPiperOrigin-RevId: 643166085","shortMessageHtmlLink":"Reduce the matrix size in testPmapMapVmapCombinations to reduce the…"}},{"before":"7cda54b0efee83907188f3f9d22ab68d645f989d","after":"8550c046828d11e7407ee27f6e0d20b1cba72683","ref":"refs/heads/test_641341474","pushedAt":"2024-06-13T23:41:53.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":"Batch `pxla.shard_args` calls triggered by `jax.device_put`\n\nWith this change, one `jax.device_put` call now corresponds to one `device_put_p.bind()` instead of one per array. Immediately, this improves the performance of `jax.device_put(...)` with a large pytree by amortizing the calls to `pxla.shard_args`. Also, backends that implement efficient batch transfers (https://github.com/tensorflow/tensorflow/pull/69096) will batch device-to-device transfers across arrays in a pytree.\n\nThe api_benchmark indicates that this CL makes `device_put` with 10 to 1000 arrays ~30% faster, likely because it reduces the number of `device_put_p.bind()` calls.\n\nPiperOrigin-RevId: 641341474","shortMessageHtmlLink":"Batch pxla.shard_args calls triggered by jax.device_put"}},{"before":"ff1cb17a63fde1949fb4035b792e8b8db35a1347","after":"bd73b98fc7c9faac94311f6f50d614cd3f95a33e","ref":"refs/heads/test_643145493","pushedAt":"2024-06-13T23:36:41.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":"Reduce the matrix size in `testPmapMapVmapCombinations` to reduce the test run time.\n\nPiperOrigin-RevId: 643145493","shortMessageHtmlLink":"Reduce the matrix size in testPmapMapVmapCombinations to reduce the…"}},{"before":"2505300b974fb8d542f24543d377996a27c97708","after":"7cda54b0efee83907188f3f9d22ab68d645f989d","ref":"refs/heads/test_641341474","pushedAt":"2024-06-13T23:36: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":"Batch `pxla.shard_args` calls triggered by `jax.device_put`\n\nWith this change, one `jax.device_put` call now corresponds to one `device_put_p.bind()` instead of one per array. Immediately, this improves the performance of `jax.device_put(...)` with a large pytree by amortizing the calls to `pxla.shard_args`. Also, backends that implement efficient batch transfers (https://github.com/tensorflow/tensorflow/pull/69096) will batch device-to-device transfers across arrays in a pytree.\n\nThe api_benchmark indicates that this CL makes `device_put` with 10 to 1000 arrays ~30% faster, likely because it reduces the number of `device_put_p.bind()` calls.\n\nPiperOrigin-RevId: 641341474","shortMessageHtmlLink":"Batch pxla.shard_args calls triggered by jax.device_put"}},{"before":"8f5f8df112530fa1f44c3b921833c7f8ece7cf31","after":"b907242f3ea1eb3da99410fcf0e33032560e3209","ref":"refs/heads/main","pushedAt":"2024-06-13T23:26:13.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 #21871 from jakevdp:fix-warnings\n\nPiperOrigin-RevId: 643156607","shortMessageHtmlLink":"Merge pull request #21871 from jakevdp:fix-warnings"}},{"before":"fa0c1ac6b2bfcce5b4a2a07d0ab40e087275f2b9","after":"2505300b974fb8d542f24543d377996a27c97708","ref":"refs/heads/test_641341474","pushedAt":"2024-06-13T23:12:34.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":"Batch `pxla.shard_args` calls triggered by `jax.device_put`\n\nWith this change, one `jax.device_put` call now corresponds to one `device_put_p.bind()` instead of one per array. Immediately, this improves the performance of `jax.device_put(...)` with a large pytree by amortizing the calls to `pxla.shard_args`. Also, backends that implement efficient batch transfers (https://github.com/tensorflow/tensorflow/pull/69096) will batch device-to-device transfers across arrays in a pytree.\n\nThe api_benchmark indicates that this CL makes `device_put` with 10 to 1000 arrays ~30% faster, likely because it reduces the number of `device_put_p.bind()` calls.\n\nPiperOrigin-RevId: 641341474","shortMessageHtmlLink":"Batch pxla.shard_args calls triggered by jax.device_put"}},{"before":"c839b268d21978c1d64de8d3f7de45ac0126a203","after":"8f5f8df112530fa1f44c3b921833c7f8ece7cf31","ref":"refs/heads/main","pushedAt":"2024-06-13T22:52:40.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 #21863 from jakevdp:dep-tracer-hash\n\nPiperOrigin-RevId: 643147305","shortMessageHtmlLink":"Merge pull request #21863 from jakevdp:dep-tracer-hash"}},{"before":null,"after":"ff1cb17a63fde1949fb4035b792e8b8db35a1347","ref":"refs/heads/test_643145493","pushedAt":"2024-06-13T22:50:07.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":"Reduce the matrix size in `testPmapMapVmapCombinations` to reduce the test run time.\n\nPiperOrigin-RevId: 643145493","shortMessageHtmlLink":"Reduce the matrix size in testPmapMapVmapCombinations to reduce the…"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"cursor":"djE6ks8AAAAEZRmdjQA","startCursor":null,"endCursor":null}},"title":"Activity · google/jax"}