{"payload":{"feedbackUrl":"https://github.com/orgs/community/discussions/53140","repo":{"id":503429735,"defaultBranch":"main","name":"praxis","ownerLogin":"google","currentUserCanPush":false,"isFork":false,"isEmpty":false,"createdAt":"2022-06-14T16:04:46.000Z","ownerAvatar":"https://avatars-ghuser.fzsz.win/u/1342004?v=4","public":true,"private":false,"isOrgOwned":true},"refInfo":{"name":"","listCacheKey":"v0:1722570523.0","currentOid":""},"activityList":{"items":[{"before":"4935bc88b0efbd634536a8587644691c8fc17d17","after":"dc3090a77dee2be767b887ecace7afd0b09bbacf","ref":"refs/heads/main","pushedAt":"2024-09-19T05:15:37.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Add unit test for binary vq.\n\nPiperOrigin-RevId: 676265751","shortMessageHtmlLink":"Add unit test for binary vq."}},{"before":"875d675ecf9d5ad26b1e38678b67ac9a4ec2c8d5","after":"4935bc88b0efbd634536a8587644691c8fc17d17","ref":"refs/heads/main","pushedAt":"2024-09-17T21:39:36.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Bump requirements.\n\nPiperOrigin-RevId: 675714694","shortMessageHtmlLink":"Bump requirements."}},{"before":"5223fac388f5c29f4b77a51da3dfc2cffe58016c","after":"875d675ecf9d5ad26b1e38678b67ac9a4ec2c8d5","ref":"refs/heads/main","pushedAt":"2024-09-12T17:11:55.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Update requirements.\n\nPiperOrigin-RevId: 673898264","shortMessageHtmlLink":"Update requirements."}},{"before":"c6636655c678ca147cf4f2316a88de024788c54d","after":"5223fac388f5c29f4b77a51da3dfc2cffe58016c","ref":"refs/heads/main","pushedAt":"2024-09-05T21:39:44.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"internal\n\nPiperOrigin-RevId: 671515254","shortMessageHtmlLink":"internal"}},{"before":"234055ade33ac690ab9a0689febeb6516db86c37","after":"c6636655c678ca147cf4f2316a88de024788c54d","ref":"refs/heads/main","pushedAt":"2024-08-30T08:37:16.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Adds streaming support for StackedTransformerRepeated.\n\nAdditionally:\n- Fixes a minor bug in the repeat layer's `init_states` function (when `unroll_in_decode` is `True`).\n- Remove input dropout from StackedTransformer streaming_step, which we are unlikely to be want during streaming inference. This is also more consistent with the extend_step of the non-streaming StackedTransformer, which doesn't include this dropout.\nPiperOrigin-RevId: 669238602","shortMessageHtmlLink":"Adds streaming support for StackedTransformerRepeated."}},{"before":"2080c2aa3deeed38cca31a34c541eec1ea68bfe5","after":"234055ade33ac690ab9a0689febeb6516db86c37","ref":"refs/heads/main","pushedAt":"2024-08-29T06:48:39.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"fix lr_schedule=None case.\n\nPiperOrigin-RevId: 668790304","shortMessageHtmlLink":"fix lr_schedule=None case."}},{"before":"cd7d76db8d5bada731ad9a30c83593c2aa0349e4","after":"2080c2aa3deeed38cca31a34c541eec1ea68bfe5","ref":"refs/heads/main","pushedAt":"2024-08-28T02:34:55.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Add vqgan losses for video models.\n\nPiperOrigin-RevId: 668237503","shortMessageHtmlLink":"Add vqgan losses for video models."}},{"before":"31a33e78447d5a6df904eb7e685fe1507aba42c9","after":"cd7d76db8d5bada731ad9a30c83593c2aa0349e4","ref":"refs/heads/main","pushedAt":"2024-08-26T20:44:35.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"adding a flag to return logits in the outputs\n\nPiperOrigin-RevId: 667700864","shortMessageHtmlLink":"adding a flag to return logits in the outputs"}},{"before":"df82a7ef58101d4eb53628ba192ae52017c1fa39","after":"31a33e78447d5a6df904eb7e685fe1507aba42c9","ref":"refs/heads/main","pushedAt":"2024-08-22T16:06:56.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Add local window attention to StackedTransformer.\n\nPiperOrigin-RevId: 666372352","shortMessageHtmlLink":"Add local window attention to StackedTransformer."}},{"before":"f7c5e7ffc81a2a57182eaa3a5666742f50e7c6c3","after":"df82a7ef58101d4eb53628ba192ae52017c1fa39","ref":"refs/heads/main","pushedAt":"2024-08-21T20:45:41.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"[pmap no rank reduce cleanup]: Update pax py_utils.py to respect rank\nreduction.\n\nPiperOrigin-RevId: 666015922","shortMessageHtmlLink":"[pmap no rank reduce cleanup]: Update pax py_utils.py to respect rank"}},{"before":"6b7edb2debbd8966e08dd33099baf925d758489b","after":"f7c5e7ffc81a2a57182eaa3a5666742f50e7c6c3","ref":"refs/heads/main","pushedAt":"2024-08-21T08:24:03.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Makes the dtype of intermediate attention masks the same as `atten_mask`. This avoids the unnecessary memory usage of float32 when `atten_mask` is bfloat16.\n\nPiperOrigin-RevId: 665756973","shortMessageHtmlLink":"Makes the dtype of intermediate attention masks the same as `atten_ma…"}},{"before":"67637df291a4de792e25a556dbfe89781ce5b7ab","after":"6b7edb2debbd8966e08dd33099baf925d758489b","ref":"refs/heads/main","pushedAt":"2024-08-20T17:38:30.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Remove unused imports via `ruff check --fix --select=F401 [paths]`, skipping\n`pytypes.py` which includes unused types by design. No functional changes.\n\nPiperOrigin-RevId: 665427368","shortMessageHtmlLink":"Remove unused imports via <code>ruff check --fix --select=F401 [paths]</code>, s…"}},{"before":"78839a5841b9971ae859d124f8e6456770960dd2","after":"ffa82e9bfa6530f96f6f9efc441ba74896693634","ref":"refs/heads/dependabot/pip/praxis/pip_package/pip-31aed340c4","pushedAt":"2024-08-20T17:27:20.000Z","pushType":"force_push","commitsCount":0,"pusher":{"login":"dependabot[bot]","name":null,"path":"/apps/dependabot","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/29110?s=80&v=4"},"commit":{"message":"Bump tensorflow\n\nBumps the pip group with 1 update in the /praxis/pip_package directory: [tensorflow](https://github.com/tensorflow/tensorflow).\n\n\nUpdates `tensorflow` from 2.9.3 to 2.12.1\n- [Release notes](https://github.com/tensorflow/tensorflow/releases)\n- [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)\n- [Commits](https://github.com/tensorflow/tensorflow/compare/v2.9.3...v2.12.1)\n\n---\nupdated-dependencies:\n- dependency-name: tensorflow\n  dependency-type: direct:production\n  dependency-group: pip\n...\n\nSigned-off-by: dependabot[bot] <support@github.com>","shortMessageHtmlLink":"Bump tensorflow"}},{"before":"439dd35f0d951a94be6bb50d6163d3a3167e84a5","after":"67637df291a4de792e25a556dbfe89781ce5b7ab","ref":"refs/heads/main","pushedAt":"2024-08-20T17:26:13.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Update requirements\n\nPiperOrigin-RevId: 665421460","shortMessageHtmlLink":"Update requirements"}},{"before":"f777c5761f02d5536146257827a8cb056f27b7b8","after":"439dd35f0d951a94be6bb50d6163d3a3167e84a5","ref":"refs/heads/main","pushedAt":"2024-08-20T08:50:09.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Add support for local windowed attention to MQA.\n\nPiperOrigin-RevId: 665237132","shortMessageHtmlLink":"Add support for local windowed attention to MQA."}},{"before":"5d55699bf75a811668b68da668ceb7b2112ccc97","after":"f777c5761f02d5536146257827a8cb056f27b7b8","ref":"refs/heads/main","pushedAt":"2024-08-19T18:44:04.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Implement videogvt magvit_v2 discriminator in praxis.\n\nPiperOrigin-RevId: 664905919","shortMessageHtmlLink":"Implement videogvt magvit_v2 discriminator in praxis."}},{"before":"e036c6dd4bf7d4f30b896d636418038669e36cc4","after":"5d55699bf75a811668b68da668ceb7b2112ccc97","ref":"refs/heads/main","pushedAt":"2024-08-14T19:33:54.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Automated Code Change\n\nPiperOrigin-RevId: 663016159","shortMessageHtmlLink":"Automated Code Change"}},{"before":"f99d63ceedf509d2caa26f85ce57c8dee517d887","after":"e036c6dd4bf7d4f30b896d636418038669e36cc4","ref":"refs/heads/main","pushedAt":"2024-08-14T18:52:39.000Z","pushType":"push","commitsCount":5,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #83 from wenscarl:pp_fix\n\nPiperOrigin-RevId: 663002539","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2445343740\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/praxis/issues/83\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/praxis/pull/83/hovercard\" href=\"https://github.com/google/praxis/pull/83\">#83</a> from wenscarl:pp_fix"}},{"before":"81154d8ea201e61e7c415568bac9b67c216f3af2","after":"f99d63ceedf509d2caa26f85ce57c8dee517d887","ref":"refs/heads/main","pushedAt":"2024-08-07T17:28:00.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Fix quantizer base dtype\n\nPiperOrigin-RevId: 660442990","shortMessageHtmlLink":"Fix quantizer base dtype"}},{"before":"84287882286362f4f5f3cce15c90276104ddce9f","after":"81154d8ea201e61e7c415568bac9b67c216f3af2","ref":"refs/heads/main","pushedAt":"2024-08-05T16:39:49.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Equivalent videogvt magvit v2 tokenizer in praxis.\n\nPiperOrigin-RevId: 659582128","shortMessageHtmlLink":"Equivalent videogvt magvit v2 tokenizer in praxis."}},{"before":null,"after":"78839a5841b9971ae859d124f8e6456770960dd2","ref":"refs/heads/dependabot/pip/praxis/pip_package/pip-31aed340c4","pushedAt":"2024-08-02T03:48:43.000Z","pushType":"branch_creation","commitsCount":0,"pusher":{"login":"dependabot[bot]","name":null,"path":"/apps/dependabot","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/29110?s=80&v=4"},"commit":{"message":"Bump tensorflow\n\nBumps the pip group with 1 update in the /praxis/pip_package directory: [tensorflow](https://github.com/tensorflow/tensorflow).\n\n\nUpdates `tensorflow` from 2.9.3 to 2.12.1\n- [Release notes](https://github.com/tensorflow/tensorflow/releases)\n- [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)\n- [Commits](https://github.com/tensorflow/tensorflow/compare/v2.9.3...v2.12.1)\n\n---\nupdated-dependencies:\n- dependency-name: tensorflow\n  dependency-type: direct:production\n  dependency-group: pip\n...\n\nSigned-off-by: dependabot[bot] <support@github.com>","shortMessageHtmlLink":"Bump tensorflow"}},{"before":"553b415cfd14553f04a339a3099e89a4fa88d915","after":null,"ref":"refs/heads/dependabot/pip/praxis/pip_package/pip-78baf76207","pushedAt":"2024-08-02T03:48:41.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"dependabot[bot]","name":null,"path":"/apps/dependabot","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/29110?s=80&v=4"}},{"before":"da4fe8dfc762e510a26c4c241f319b38c1f34366","after":"84287882286362f4f5f3cce15c90276104ddce9f","ref":"refs/heads/main","pushedAt":"2024-08-02T03:47:37.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Update requirements with new jaxlib version.\n\nPiperOrigin-RevId: 658641176","shortMessageHtmlLink":"Update requirements with new jaxlib version."}},{"before":"5b5ed76dfda68e1fbc2d53499dc7c2ae8091bc15","after":"da4fe8dfc762e510a26c4c241f319b38c1f34366","ref":"refs/heads/main","pushedAt":"2024-07-24T17:26:36.000Z","pushType":"push","commitsCount":5,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #80 from kaixih:fix_fp8_einsum\n\nPiperOrigin-RevId: 655615204","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2399281167\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/praxis/issues/80\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/praxis/pull/80/hovercard\" href=\"https://github.com/google/praxis/pull/80\">#80</a> from kaixih:fix_fp8_einsum"}},{"before":"e9886a34a66bf60e37f8b8bfa62fdef24f2769e2","after":"5b5ed76dfda68e1fbc2d53499dc7c2ae8091bc15","ref":"refs/heads/main","pushedAt":"2024-07-24T09:43:08.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"[DotProductAttention] When relative bias is used, set query_segment_pos and key_segment_pos to default positions (jnp.arange) if not provided.\n\nPiperOrigin-RevId: 655490645","shortMessageHtmlLink":"[DotProductAttention] When relative bias is used, set query_segment_p…"}},{"before":"25b586fa1598072d5e15f4601f632d121a7e701b","after":"e9886a34a66bf60e37f8b8bfa62fdef24f2769e2","ref":"refs/heads/main","pushedAt":"2024-07-24T05:54:54.000Z","pushType":"push","commitsCount":6,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #40 from kaixih:fp8_meta_custom_grad_accumulation\n\nPiperOrigin-RevId: 655437010","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2084667370\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/praxis/issues/40\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/praxis/pull/40/hovercard\" href=\"https://github.com/google/praxis/pull/40\">#40</a> from kaixih:fp8_meta_custom_grad_accumulation"}},{"before":"eb1601fa0cd52e5981c61c4b355dfe104c5f5527","after":"25b586fa1598072d5e15f4601f632d121a7e701b","ref":"refs/heads/main","pushedAt":"2024-07-23T03:15:30.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Add missing docs and attribute types for `AdamOptState` and `_ShardedAdamHelper`\n\nPiperOrigin-RevId: 654994459","shortMessageHtmlLink":"Add missing docs and attribute types for <code>AdamOptState</code> and `_Sharded…"}},{"before":"43fe5640f005242f32f2ea96a8053a8f2f4d8efa","after":"eb1601fa0cd52e5981c61c4b355dfe104c5f5527","ref":"refs/heads/main","pushedAt":"2024-07-22T16:38:20.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Move default values for `momentum` and `nesterov` from comments to function signature, making them optional as documented. Previously, they were required, even\nthough the documentation implied they had default values, and hence were thought to\nbe optional, but the default values were actually never used.\n\nAlso capitalize \"Nesterov\" since it's a proper name.\n\nPiperOrigin-RevId: 654787611","shortMessageHtmlLink":"Move default values for <code>momentum</code> and <code>nesterov</code> from comments to fu…"}},{"before":"eb5473d67342470ef769dfc0bfbad425ee480857","after":"43fe5640f005242f32f2ea96a8053a8f2f4d8efa","ref":"refs/heads/main","pushedAt":"2024-07-17T05:51:19.000Z","pushType":"push","commitsCount":2,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"Merge pull request #53 from kaixih:cudnn_attention_dev\n\nPiperOrigin-RevId: 653100646","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2203300144\" data-permission-text=\"Title is private\" data-url=\"https://github.com/google/praxis/issues/53\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/google/praxis/pull/53/hovercard\" href=\"https://github.com/google/praxis/pull/53\">#53</a> from kaixih:cudnn_attention_dev"}},{"before":"dafadc1f922dc0cf422f44baf4922c9f0dfe4c15","after":"eb5473d67342470ef769dfc0bfbad425ee480857","ref":"refs/heads/main","pushedAt":"2024-07-16T21:01:38.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"copybara-service[bot]","name":null,"path":"/apps/copybara-service","primaryAvatarUrl":"https://avatars-ghuser.fzsz.win/in/44061?s=80&v=4"},"commit":{"message":"added more arguments to the transformer quantizer\n\nfixed assert statement in attentions.py to allow training with quantizer\n\nPiperOrigin-RevId: 652968446","shortMessageHtmlLink":"added more arguments to the transformer quantizer"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"cursor":"djE6ks8AAAAEum3V6wA","startCursor":null,"endCursor":null}},"title":"Activity · google/praxis"}