{"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:1717223799.0","currentOid":""},"activityList":{"items":[{"before":null,"after":"eb42879b72a32424952675f6ae6ebcc56986bd60","ref":"refs/heads/test_639262876","pushedAt":"2024-06-01T06:36:39.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":"[Mosaic] Always use vector native tiling when loading from memory that has a single column of tiles\n\nPiperOrigin-RevId: 639262876","shortMessageHtmlLink":"[Mosaic] Always use vector native tiling when loading from memory tha…"}},{"before":"befec2c494df482601fe32ca66d63445cd277cbf","after":"aaded57ef8cad72a6648e8f0b2d334931676ff11","ref":"refs/heads/test_639203738","pushedAt":"2024-06-01T06:04: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":"Port Cholesky Factorization to XLA's FFI\n\nPiperOrigin-RevId: 639203738","shortMessageHtmlLink":"Port Cholesky Factorization to XLA's FFI"}},{"before":null,"after":"befec2c494df482601fe32ca66d63445cd277cbf","ref":"refs/heads/test_639203738","pushedAt":"2024-06-01T05:57:52.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":"Port Cholesky Factorization to XLA's FFI\n\nPiperOrigin-RevId: 639203738","shortMessageHtmlLink":"Port Cholesky Factorization to XLA's FFI"}},{"before":"432159a9d315b9b9203064f9c635ecd1916593af","after":"be1e40dc2e1777d83b870afa31e178123f2a1366","ref":"refs/heads/main","pushedAt":"2024-06-01T03:40: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":"Copybara import of the project:\n\n--\nf79d1060cccf7c9a1c02d0bcab06c6ee0ef795a8 by George Necula <gcnecula@gmail.com>:\n\n[export] Fix\n\nA user reported an error when trying to export a function\nthat has a \"lower\" attribute (to impersonate a jitted function)\nbut does not have a \"__name__\" attribute.\nThe solution is to use the default name \"<unnamed function>\".\n\nWhile I was at it I have added a `util.fun_name` to get\nthe name of a Callable, and I use it in several places.\n\nCOPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/21572 from gnecula:exp_fix_name f79d1060cccf7c9a1c02d0bcab06c6ee0ef795a8\nPiperOrigin-RevId: 639236990","shortMessageHtmlLink":"Copybara import of the project:"}},{"before":"30845b524918bc36ac480dc3c2b3f68fe619cc7f","after":"eb8c10efd145824bc497658c25fcfc4da35d5ffb","ref":"refs/heads/test_639227607","pushedAt":"2024-06-01T03:10: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":"Fixed the logging due to a recent change.\n\nPiperOrigin-RevId: 639227607","shortMessageHtmlLink":"Fixed the logging due to a recent change."}},{"before":null,"after":"30845b524918bc36ac480dc3c2b3f68fe619cc7f","ref":"refs/heads/test_639227607","pushedAt":"2024-06-01T02:59:46.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":"Fixed the logging due to a recent change.\n\nPiperOrigin-RevId: 639227607","shortMessageHtmlLink":"Fixed the logging due to a recent change."}},{"before":"dfd7d17c1d6f51f651839aff462ae94edefc7dc7","after":"432159a9d315b9b9203064f9c635ecd1916593af","ref":"refs/heads/main","pushedAt":"2024-06-01T02:55: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 XLA dependency to use revision\nhttp://github.com/openxla/xla/commit/ea480cecc0df5ead6767163b57a96b102d367408.\n\nPiperOrigin-RevId: 639227230","shortMessageHtmlLink":"Update XLA dependency to use revision"}},{"before":"dfd7d17c1d6f51f651839aff462ae94edefc7dc7","after":null,"ref":"refs/heads/test_639150552","pushedAt":"2024-06-01T01:15: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":"1edb94ec464d7e9965952ab2e3450f3dc8ecf57a","after":"dfd7d17c1d6f51f651839aff462ae94edefc7dc7","ref":"refs/heads/main","pushedAt":"2024-06-01T01:15: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":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is more efficient than tile_assignment_devices.\n\nHloSharding V1 -> HloSharding V2.\n\nPiperOrigin-RevId: 639210975","shortMessageHtmlLink":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is…"}},{"before":"4d13a9d68a54d8c55e84a86822b50cf978fc91fb","after":"dfd7d17c1d6f51f651839aff462ae94edefc7dc7","ref":"refs/heads/test_639150552","pushedAt":"2024-06-01T01:15:02.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":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is more efficient than tile_assignment_devices.\n\nHloSharding V1 -> HloSharding V2.\n\nPiperOrigin-RevId: 639210975","shortMessageHtmlLink":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is…"}},{"before":"49403cf60ae38071337cb00fc45470c61a1c4770","after":"4d13a9d68a54d8c55e84a86822b50cf978fc91fb","ref":"refs/heads/test_639150552","pushedAt":"2024-06-01T00:56:59.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":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is more efficient than tile_assignment_devices.\n\nHloSharding V1 -> HloSharding V2.\n\nPiperOrigin-RevId: 639150552","shortMessageHtmlLink":"[JAX] Use iota_reshape_dims and iota_transpose_perm in pxla, which is…"}},{"before":"1edb94ec464d7e9965952ab2e3450f3dc8ecf57a","after":null,"ref":"refs/heads/test_636764537","pushedAt":"2024-06-01T00:51:30.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":"5eb679ee344187079ee3863632a4b15247a5dc36","after":"1edb94ec464d7e9965952ab2e3450f3dc8ecf57a","ref":"refs/heads/main","pushedAt":"2024-06-01T00:51:29.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":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+\n\nNeeded a little more backfill for TPU load\n\nPiperOrigin-RevId: 639206243","shortMessageHtmlLink":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+"}},{"before":"075e0c8349ed085aedf04e3c3f5589b49f0f509a","after":"1edb94ec464d7e9965952ab2e3450f3dc8ecf57a","ref":"refs/heads/test_636764537","pushedAt":"2024-06-01T00:51:28.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":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+\n\nNeeded a little more backfill for TPU load\n\nPiperOrigin-RevId: 639206243","shortMessageHtmlLink":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+"}},{"before":"53bbb662f7ae5ec0b29e561581942733ba553ecd","after":"075e0c8349ed085aedf04e3c3f5589b49f0f509a","ref":"refs/heads/test_636764537","pushedAt":"2024-06-01T00:35:28.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":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+\n\nNeeded a little more backfill for TPU load\n\nPiperOrigin-RevId: 636764537","shortMessageHtmlLink":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+"}},{"before":null,"after":"245ed5ee45f076ef59baec0955f77ca715a7918d","ref":"refs/heads/test_639193156","pushedAt":"2024-05-31T23:52:04.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":"chore: adopt new local wheel installation logic\n\nPiperOrigin-RevId: 639193156","shortMessageHtmlLink":"chore: adopt new local wheel installation logic"}},{"before":"cb07ba4c5d25b9f3f318a57772eeff1134ef2992","after":null,"ref":"refs/heads/test_637803299","pushedAt":"2024-05-31T23:37:55.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":"d6894b6a9c2853d01c9afc403adbdb938aafdfe9","after":"0213e12c81de8a591227fed1eb51ddc7c35b9bbf","ref":"refs/heads/test_620111786","pushedAt":"2024-05-31T23:35:59.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":"Port JAX LAPACK custom calls to XLA's FFI (API v4)\n\n- Upgrade CPU custom calls to FFI.\n- Fix the array shape bug in the Tridiagonal Reduction kernel (insufficient allocation, shape off by one).\n- Correct and extend the `_lapack` module stub to not produce warnings.\n- Make Schur Decomposition return a fixed number of values.\n- (WiP) Update the implementation of Schur form to consider the `select` parameter.\n\nPiperOrigin-RevId: 620111786","shortMessageHtmlLink":"Port JAX LAPACK custom calls to XLA's FFI (API v4)"}},{"before":"5eb679ee344187079ee3863632a4b15247a5dc36","after":null,"ref":"refs/heads/test_639134828","pushedAt":"2024-05-31T23:35:54.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":"c8207ee8318781925e6d3d16e415d82aefaf4039","after":"5eb679ee344187079ee3863632a4b15247a5dc36","ref":"refs/heads/main","pushedAt":"2024-05-31T23:35:52.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":"Hypothesis isn't compatible with subTest and jax tests sometimes report these\nwarnings as errors. Disable to allow these tests to pass.\n\nPiperOrigin-RevId: 639189411","shortMessageHtmlLink":"Hypothesis isn't compatible with subTest and jax tests sometimes repo…"}},{"before":"f18ea0bca90691eb1e66de3ea0ee414a289b00e7","after":"5eb679ee344187079ee3863632a4b15247a5dc36","ref":"refs/heads/test_639134828","pushedAt":"2024-05-31T23:35:51.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":"Hypothesis isn't compatible with subTest and jax tests sometimes report these\nwarnings as errors. Disable to allow these tests to pass.\n\nPiperOrigin-RevId: 639189411","shortMessageHtmlLink":"Hypothesis isn't compatible with subTest and jax tests sometimes repo…"}},{"before":"31ee321178556fd9c86d347fca2ed94e06e0daa3","after":"d6894b6a9c2853d01c9afc403adbdb938aafdfe9","ref":"refs/heads/test_620111786","pushedAt":"2024-05-31T23:23:12.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":"Port JAX LAPACK custom calls to XLA's FFI (API v4)\n\n- Upgrade CPU custom calls to FFI.\n- Fix the array shape bug in the Tridiagonal Reduction kernel (insufficient allocation, shape off by one).\n- Correct and extend the `_lapack` module stub to not produce warnings.\n- Make Schur Decomposition return a fixed number of values.\n- (WiP) Update the implementation of Schur form to consider the `select` parameter.\n\nPiperOrigin-RevId: 620111786","shortMessageHtmlLink":"Port JAX LAPACK custom calls to XLA's FFI (API v4)"}},{"before":"dfc51c81ab71cc1b77e0e19f15811b1d5814a88a","after":"f18ea0bca90691eb1e66de3ea0ee414a289b00e7","ref":"refs/heads/test_639134828","pushedAt":"2024-05-31T23:22:23.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":"Hypothesis isn't compatible with subTest and jax tests sometimes report these\nwarnings as errors. Disable to allow these tests to pass.\n\nPiperOrigin-RevId: 639134828","shortMessageHtmlLink":"Hypothesis isn't compatible with subTest and jax tests sometimes repo…"}},{"before":"80c1464a005f95ada0de88924b1e2d832e96b9f9","after":"31ee321178556fd9c86d347fca2ed94e06e0daa3","ref":"refs/heads/test_620111786","pushedAt":"2024-05-31T22:54:40.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":"Port JAX LAPACK custom calls to XLA's FFI (API v4)\n\n- Upgrade CPU custom calls to FFI.\n- Fix the array shape bug in the Tridiagonal Reduction kernel (insufficient allocation, shape off by one).\n- Correct and extend the `_lapack` module stub to not produce warnings.\n- Make Schur Decomposition return a fixed number of values.\n- (WiP) Update the implementation of Schur form to consider the `select` parameter.\n\nPiperOrigin-RevId: 620111786","shortMessageHtmlLink":"Port JAX LAPACK custom calls to XLA's FFI (API v4)"}},{"before":"864f4e559103a9ad3388121a9eb7b2a73b82aaa0","after":"80c1464a005f95ada0de88924b1e2d832e96b9f9","ref":"refs/heads/test_620111786","pushedAt":"2024-05-31T22:32: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":"Port JAX LAPACK custom calls to XLA's FFI (API v4)\n\n- Upgrade CPU custom calls to FFI.\n- Fix the array shape bug in the Tridiagonal Reduction kernel (insufficient allocation, shape off by one).\n- Correct and extend the `_lapack` module stub to not produce warnings.\n- Make Schur Decomposition return a fixed number of values.\n- (WiP) Update the implementation of Schur form to consider the `select` parameter.\n\nPiperOrigin-RevId: 620111786","shortMessageHtmlLink":"Port JAX LAPACK custom calls to XLA's FFI (API v4)"}},{"before":"0591620932000b24e170fa1d1686d4baf11245ed","after":"c8207ee8318781925e6d3d16e415d82aefaf4039","ref":"refs/heads/main","pushedAt":"2024-05-31T22:28: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 #21568 from mattjj:custom-vjp-bwd-produce-list-or-tuple\n\nPiperOrigin-RevId: 639170346","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2328633299\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/jax/issues/21568\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/jax/pull/21568/hovercard\" href=\"https://github.com/google/jax/pull/21568\">#21568</a> from mattjj:custom-vjp-bwd-produce-list-or-…"}},{"before":"50661c34956690f96e45629f4c936edc76225743","after":"864f4e559103a9ad3388121a9eb7b2a73b82aaa0","ref":"refs/heads/test_620111786","pushedAt":"2024-05-31T22:17: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":"Port JAX LAPACK custom calls to XLA's FFI (API v4)\n\n- Upgrade CPU custom calls to FFI.\n- Fix the array shape bug in the Tridiagonal Reduction kernel (insufficient allocation, shape off by one).\n- Correct and extend the `_lapack` module stub to not produce warnings.\n- Make Schur Decomposition return a fixed number of values.\n- (WiP) Update the implementation of Schur form to consider the `select` parameter.\n\nPiperOrigin-RevId: 620111786","shortMessageHtmlLink":"Port JAX LAPACK custom calls to XLA's FFI (API v4)"}},{"before":"e81ae492cab1f879a4a682eb59598db0b1e530d9","after":"dfc51c81ab71cc1b77e0e19f15811b1d5814a88a","ref":"refs/heads/test_639134828","pushedAt":"2024-05-31T22:16:11.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":"Hypothesis isn't compatible with subTest and jax tests sometimes report these\nwarnings as errors. Disable to allow these tests to pass.\n\nPiperOrigin-RevId: 639134828","shortMessageHtmlLink":"Hypothesis isn't compatible with subTest and jax tests sometimes repo…"}},{"before":null,"after":"53edef85ab1fb516a4bd1e30403fd2fbcb5cf8d7","ref":"refs/heads/test_639162448","pushedAt":"2024-05-31T22:05:56.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":"[jax:cpu] Add a test config to run JAX with new XLA:CPU runtime\n\nXLA:CPU is migrating from compiling monolithic LLVM function for the whole HLO module to a thin runtime with separate functions for each kernel (fusion, individual operation, library call, etc.). While new runtime is not enabled by default we will use explicit opt-in on tests that are already compatible.\n\nThis tag will be removed after XLA:CPU will switch to the new runtime by default.\n\nPiperOrigin-RevId: 639162448","shortMessageHtmlLink":"[jax:cpu] Add a test config to run JAX with new XLA:CPU runtime"}},{"before":"a5ece4a407c6e1055ca8e13ee93d82a7d2e1e3c4","after":"53bbb662f7ae5ec0b29e561581942733ba553ecd","ref":"refs/heads/test_636764537","pushedAt":"2024-05-31T21:49:59.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":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+\n\nNeeded a little more backfill for TPU load\n\nPiperOrigin-RevId: 636764537","shortMessageHtmlLink":"[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"cursor":"djE6ks8AAAAEWZkeOwA","startCursor":null,"endCursor":null}},"title":"Activity · google/jax"}