OpenScientificCodeRegistry/Database

1{2 "format": "oscr-script-manifest/1",3 "repository": "github.com/google/flax",4 "url": "http://github.com/google/flax",5 "host": "github.com",6 "commit": "81b168d62f151feca9e8e7abc591df80168395b4",7 "license": "Apache-2.0",8 "license_confirmed_by": "license file LICENSE",9 "redistribution": "yes",10 "files": [11  {12   "path": ".github/analytics/get_repo_metrics.py",13   "sha256": "a99e73524c45632391386d83c2a242621312bcf7f488dc2d145af44e7c010fce",14   "language": "Python",15   "lines": 440,16   "truncated": false,17   "block": 23,18   "row": 1181719  },20  {21   "path": "LICENSE",22   "sha256": "25e8791da26f7e74c4baf19eafd7d137b64a46952ef840ed01c172b680cf4046",23   "language": "License",24   "lines": 200,25   "truncated": false,26   "block": 22,27   "row": 1416628  },29  {30   "path": "README.md",31   "sha256": "b20519acb9540abd14ba4e7ff2419e435c6bec19bcdb9ac65adcc846176943c3",32   "language": "Text",33   "lines": 172,34   "truncated": false,35   "block": 24,36   "row": 85237  },38  {39   "path": "benchmarks/nnx_graph_overhead.py",40   "sha256": "0ef2814c067fe36e59a0508283916d7da9aef37ebbc45d9d84f363037a195148",41   "language": "Python",42   "lines": 169,43   "truncated": false,44   "block": 23,45   "row": 634346  },47  {48   "path": "benchmarks/nnx_mlpmixer_training.py",49   "sha256": "4ae33d51a74894197f043d3b358383326046e03c58d2f153a82e4612612289b8",50   "language": "Python",51   "lines": 239,52   "truncated": false,53   "block": 23,54   "row": 808955  },56  {57   "path": "benchmarks/nnx_simple_training.py",58   "sha256": "8e6499e5a6e11df87fe848beb4f8e25dad338257426afce8f775d60b420c1573",59   "language": "Python",60   "lines": 260,61   "truncated": false,62   "block": 23,63   "row": 865164  },65  {66   "path": "benchmarks/nnx_state_traversal.py",67   "sha256": "ebc7d0e35521dc9fa1bef704b362f93d668b983cf3e5d988b32cd79befd25496",68   "language": "Python",69   "lines": 106,70   "truncated": false,71   "block": 23,72   "row": 381473  },74  {75   "path": "benchmarks/tracing/__init__.py",76   "sha256": "e12de960dba2c3acb33e634d9b68918b8d3f25ff719478fea0845bf46a78b319",77   "language": "Python",78   "lines": 13,79   "truncated": false,80   "block": 15,81   "row": 707482  },83  {84   "path": "benchmarks/tracing/gemma.py",85   "sha256": "9b811c93d316b4005aa8d306229809b0d124d9c78f2e6434d7aa545e3807d39d",86   "language": "Python",87   "lines": 258,88   "truncated": false,89   "block": 23,90   "row": 891191  },92  {93   "path": "benchmarks/tracing/imagenet.py",94   "sha256": "3b20963278cd64c09abdb8ed293050deadbbcc077955f99db13013ae99d688f1",95   "language": "Python",96   "lines": 255,97   "truncated": false,98   "block": 23,99   "row": 8624100  },101  {102   "path": "benchmarks/tracing/lm1b.py",103   "sha256": "c3b26010874ddb11c640f9999e8b8ae532009b94c505e4fcddc0b43ca8ba5f98",104   "language": "Python",105   "lines": 254,106   "truncated": false,107   "block": 23,108   "row": 8767109  },110  {111   "path": "benchmarks/tracing/mnist.py",112   "sha256": "35b9321f0760cf0c005db7ccf86d4d9982b59208ef5ac7bd1f388ddb44e23586",113   "language": "Python",114   "lines": 97,115   "truncated": false,116   "block": 23,117   "row": 3587118  },119  {120   "path": "benchmarks/tracing/nlp_seq.py",121   "sha256": "66058cc2a6032ee239fae51a5d15d56e4ed7e677db681191a59200484bf8d66a",122   "language": "Python",123   "lines": 242,124   "truncated": false,125   "block": 23,126   "row": 8249127  },128  {129   "path": "benchmarks/tracing/ogbg_molpcba.py",130   "sha256": "4df5f054c6287a1c677993f37b56e35ea269ce63d443fefd0dd406fb21a1e652",131   "language": "Python",132   "lines": 227,133   "truncated": false,134   "block": 23,135   "row": 8281136  },137  {138   "path": "benchmarks/tracing/ppo.py",139   "sha256": "025711b011a2dbf0e49f1e631a5b587ac57a1c67e003e206ffcc6f2e8fb15292",140   "language": "Python",141   "lines": 131,142   "truncated": false,143   "block": 23,144   "row": 4794145  },146  {147   "path": "benchmarks/tracing/run_all_benchmarks.sh",148   "sha256": "84adad879681a2f7eb18622ee12dc5df1a7f752ad0e10658b9608b59b62c558d",149   "language": "Shell",150   "lines": 26,151   "truncated": false,152   "block": 23,153   "row": 17429154  },155  {156   "path": "benchmarks/tracing/seq2seq.py",157   "sha256": "4d6684d83773ca9bf652c0188680226dcf0ed5b4a10dfe3b30b3cdf832974a0b",158   "language": "Python",159   "lines": 129,160   "truncated": false,161   "block": 23,162   "row": 4903163  },164  {165   "path": "benchmarks/tracing/sst2.py",166   "sha256": "0c0aa3f628b42138d5a55d71f7823d6fb3c418382d2bc81f26a567d3bed4b9ef",167   "language": "Python",168   "lines": 167,169   "truncated": false,170   "block": 23,171   "row": 6465172  },173  {174   "path": "benchmarks/tracing/tracing_benchmark.py",175   "sha256": "ef6feee854099e4d925dc3ff6cb1ad6a385173762a125562b076d78f9dc25825",176   "language": "Python",177   "lines": 95,178   "truncated": false,179   "block": 23,180   "row": 2970181  },182  {183   "path": "benchmarks/tracing/vae.py",184   "sha256": "7fa59f58d38ed71d0bfaf13d103eeb73ba2990ea426b888d5d1ce3d0a0a9302b",185   "language": "Python",186   "lines": 98,187   "truncated": false,188   "block": 23,189   "row": 3211190  },191  {192   "path": "benchmarks/tracing/wmt.py",193   "sha256": "1be319cb1a6b6dffffda05e90b857c9cd808e7e3e8bcc5311d4deaec452275b1",194   "language": "Python",195   "lines": 313,196   "truncated": false,197   "block": 23,198   "row": 9722199  },200  {201   "path": "docs/_ext/codediff.py",202   "sha256": "ac16f9ddf24af46e98091df06e83b8778cd15bae2a445be16e9624fbcfa3fa76",203   "language": "Python",204   "lines": 221,205   "truncated": false,206   "block": 23,207   "row": 8436208  },209  {210   "path": "docs/_ext/codediff_test.py",211   "sha256": "8b8717cbeba50010ccf8806b612c672108c53c4c301fb05e4af90afb5a26fafe",212   "language": "Python",213   "lines": 120,214   "truncated": false,215   "block": 23,216   "row": 5152217  },218  {219   "path": "docs/_ext/flax_module.py",220   "sha256": "4a18f56f4ab83dee3cb1344b17985924374241595c9739a8fb353bb8008b1ba3",221   "language": "Python",222   "lines": 87,223   "truncated": false,224   "block": 23,225   "row": 2039226  },227  {228   "path": "docs/conf.py",229   "sha256": "3cb786fe0fe21d76a0c1aabbdb39c996acb1c7bbbf1cb762ad9a2366e0c5bd3c",230   "language": "Python",231   "lines": 190,232   "truncated": false,233   "block": 23,234   "row": 7648235  },236  {237   "path": "docs/conf_sphinx_patch.py",238   "sha256": "3425e15890f250bb2fcc2df062c75a2001e28fe3aef36352b49c2002d4fd9556",239   "language": "Python",240   "lines": 200,241   "truncated": false,242   "block": 23,243   "row": 7961244  },245  {246   "path": "docs/guides/data_preprocessing/loading_datasets.ipynb",247   "sha256": "37c7960bf1fb7ccbad8c136cfee64b579e8303d7484ed1315c92a286a3666f8e",248   "language": "Jupyter",249   "lines": 120,250   "truncated": false,251   "block": 22,252   "row": 12801253  },254  {255   "path": "docs/guides/flax_fundamentals/flax_basics.ipynb",256   "sha256": "7ae4498b7e756a00bff12cfecc69abee7261a5683d0084be94018d38a38890fa",257   "language": "Jupyter",258   "lines": 424,259   "truncated": false,260   "block": 22,261   "row": 13582262  },263  {264   "path": "docs/guides/flax_fundamentals/rng_guide.ipynb",265   "sha256": "c0566249742741d88f601d3041de337805e8f1cbb4fe88c09d1149eb960bedac",266   "language": "Jupyter",267   "lines": 688,268   "truncated": false,269   "block": 22,270   "row": 13672271  },272  {273   "path": "docs/guides/flax_sharp_bits.ipynb",274   "sha256": "66c5d267020bba24d3a8cda70c04fb35561534ef32704a6777355adf947063f4",275   "language": "Jupyter",276   "lines": 87,277   "truncated": false,278   "block": 22,279   "row": 12919280  },281  {282   "path": "docs/guides/model_inspection/model_surgery.ipynb",283   "sha256": "1ddc1737b57e2e1066777bb36e1947a4e181119dfbe1c69059587009094b150a",284   "language": "Jupyter",285   "lines": 123,286   "truncated": false,287   "block": 22,288   "row": 12777289  },290  {291   "path": "docs/guides/parallel_training/flax_on_pjit.ipynb",292   "sha256": "475b0b18fcee845c2b818b10567228ffd729f14401d4e688626fb65b0656740b",293   "language": "Jupyter",294   "lines": 458,295   "truncated": false,296   "block": 22,297   "row": 13638298  },299  {300   "path": "docs/guides/quantization/fp8_basics.ipynb",301   "sha256": "4738defc4a85fd07b7d8b4b1ec922158fc5e0d6d93f7df6015fdd3916a23c030",302   "language": "Jupyter",303   "lines": 346,304   "truncated": false,305   "block": 22,306   "row": 13387307  },308  {309   "path": "docs/guides/training_techniques/transfer_learning.ipynb",310   "sha256": "943b516f8f342462b5f369472e7bc81e3a0b74a0c85f137c8ed27256ed7542b1",311   "language": "Jupyter",312   "lines": 163,313   "truncated": false,314   "block": 22,315   "row": 13066316  },317  {318   "path": "docs/guides/training_techniques/use_checkpointing.ipynb",319   "sha256": "95ac3faf2d6bff74bde44db9061e4f25c63014e6f3bcf7f27259ccb1653ced99",320   "language": "Jupyter",321   "lines": 476,322   "truncated": false,323   "block": 22,324   "row": 13651325  },326  {327   "path": "docs/linen_intro.ipynb",328   "sha256": "96999e9c0c80e0af4c29edb28f69dd1b1602eef54fd0fe648027837b1e84ba6f",329   "language": "Jupyter",330   "lines": 536,331   "truncated": false,332   "block": 22,333   "row": 13586334  },335  {336   "path": "docs/quick_start.ipynb",337   "sha256": "6f48b81b61639506e0739c881589e082bac8e9ccbec3f5ebc613bf9917ee7dfa",338   "language": "Jupyter",339   "lines": 326,340   "truncated": false,341   "block": 22,342   "row": 13412343  },344  {345   "path": "docs_nnx/_ext/codediff.py",346   "sha256": "ac16f9ddf24af46e98091df06e83b8778cd15bae2a445be16e9624fbcfa3fa76",347   "language": "Python",348   "lines": 221,349   "truncated": false,350   "block": 23,351   "row": 8436352  },353  {354   "path": "docs_nnx/_ext/codediff_test.py",355   "sha256": "8b8717cbeba50010ccf8806b612c672108c53c4c301fb05e4af90afb5a26fafe",356   "language": "Python",357   "lines": 120,358   "truncated": false,359   "block": 23,360   "row": 5152361  },362  {363   "path": "docs_nnx/_ext/flax_module.py",364   "sha256": "4a18f56f4ab83dee3cb1344b17985924374241595c9739a8fb353bb8008b1ba3",365   "language": "Python",366   "lines": 87,367   "truncated": false,368   "block": 23,369   "row": 2039370  },371  {372   "path": "docs_nnx/conf.py",373   "sha256": "56a089d64f494c767e77a388a7f393d449eaf941e1741dd294a76dac6359a978",374   "language": "Python",375   "lines": 195,376   "truncated": false,377   "block": 23,378   "row": 7875379  },380  {381   "path": "docs_nnx/conf_sphinx_patch.py",382   "sha256": "3425e15890f250bb2fcc2df062c75a2001e28fe3aef36352b49c2002d4fd9556",383   "language": "Python",384   "lines": 200,385   "truncated": false,386   "block": 23,387   "row": 7961388  },389  {390   "path": "docs_nnx/examples/digits_diffusion_model.ipynb",391   "sha256": "11d8845035b947b1b7eae19fe7cd60df32ccc87b393b16736398ff1f0b1da41e",392   "language": "Jupyter",393   "lines": 428,394   "truncated": false,395   "block": 22,396   "row": 13570397  },398  {399   "path": "docs_nnx/examples/gemma.ipynb",400   "sha256": "74e6c3a34507b0cb0ef79b420bf028cb4b9ee90def4d296f9a70c7b522a9850b",401   "language": "Jupyter",402   "lines": 145,403   "truncated": false,404   "block": 22,405   "row": 13008406  },407  {408   "path": "docs_nnx/examples/image_segmentation.ipynb",409   "sha256": "dd832a58ff4a3f3111a29aff679a7cf156c2b8bdaead6f0bb0596bbe08e43107",410   "language": "Jupyter",411   "lines": 1392,412   "truncated": false,413   "block": 22,414   "row": 13772415  },416  {417   "path": "docs_nnx/examples/machine_translation.ipynb",418   "sha256": "014768f6f6af0a1aeb5e29200dba6e72eda344def9c61c0f966a92974c16a561",419   "language": "Jupyter",420   "lines": 566,421   "truncated": false,422   "block": 22,423   "row": 13641424  },425  {426   "path": "docs_nnx/examples/minigpt.ipynb",427   "sha256": "e5731e764f5ab78edb8aaeb00b7ea769c03b3e63af2c10936da67524bc694137",428   "language": "Jupyter",429   "lines": 490,430   "truncated": false,431   "block": 22,432   "row": 13592433  },434  {435   "path": "docs_nnx/examples/object_detection_detr.ipynb",436   "sha256": "2c51246c2025be3495c7d427bbc8f24ad4d907eea9309a4caef65d209161c3ac",437   "language": "Jupyter",438   "lines": 298,439   "truncated": true,440   "block": 22,441   "row": 13841442  },443  {444   "path": "docs_nnx/examples/vit_training.ipynb",445   "sha256": "9f2019786318945371a0519537fcf19c0e65022f3ec01d42c7277b9af7306f6f",446   "language": "Jupyter",447   "lines": 870,448   "truncated": false,449   "block": 22,450   "row": 13710451  },452  {453   "path": "docs_nnx/guides/bridge_guide.ipynb",454   "sha256": "7de7a2dd0a152986ae30eb0a1924c8d2e118932c14fd574bdd95f184b15af430",455   "language": "Jupyter",456   "lines": 480,457   "truncated": false,458   "block": 22,459   "row": 13597460  },461  {462   "path": "docs_nnx/guides/checkpointing.ipynb",463   "sha256": "bc8d6d64e039f2c44ddb2b7cec683ab9c9d33880922284048d016592032c881a",464   "language": "Jupyter",465   "lines": 208,466   "truncated": false,467   "block": 22,468   "row": 13829469  },470  {471   "path": "docs_nnx/guides/data_loaders.ipynb",472   "sha256": "86776da593ea38feb4dd2bbc778444cb9dc0d08a100966a767f47d5860ca4ddd",473   "language": "Jupyter",474   "lines": 187,475   "truncated": false,476   "block": 22,477   "row": 13122478  },479  {480   "path": "docs_nnx/guides/demo.ipynb",481   "sha256": "9d75f68232d2ba183d50e211640dd3f7c0c8ac7518cb5eed36c211c829c22e4d",482   "language": "Jupyter",483   "lines": 144,484   "truncated": false,485   "block": 22,486   "row": 12726487  },488  {489   "path": "docs_nnx/guides/extracting_intermediates.ipynb",490   "sha256": "4c655111ff70c4c19c6e6f6b1945355ad84b93d119ac203d470db4baea657ccb",491   "language": "Jupyter",492   "lines": 250,493   "truncated": false,494   "block": 22,495   "row": 13345496  },497  {498   "path": "docs_nnx/guides/filters_guide.ipynb",499   "sha256": "fc05dfec9f5207cf2b3721ec1a38225e55e8578b175197555e6c71b9c10a0a07",500   "language": "Jupyter",501   "lines": 190,502   "truncated": false,503   "block": 22,504   "row": 13185505  },506  {507   "path": "docs_nnx/guides/flax_gspmd.ipynb",508   "sha256": "4dc909331709226fc6b167f5850cc3da64ebb4adc7b7fb1b4e4a422f2927818b",509   "language": "Jupyter",510   "lines": 396,511   "truncated": false,512   "block": 22,513   "row": 13561514  },515  {516   "path": "docs_nnx/guides/optimization_cookbook.ipynb",517   "sha256": "fee7f5b07d2d0bff1094a59335e67af0f767fc5a05df6c92b689cdc727496fe5",518   "language": "Jupyter",519   "lines": 220,520   "truncated": false,521   "block": 22,522   "row": 13032523  },524  {525   "path": "docs_nnx/guides/performance.ipynb",526   "sha256": "2702ad9e6e82b5be84854c19b2cf674ef3f290fb1460504c078339e6e7e14478",527   "language": "Jupyter",528   "lines": 117,529   "truncated": false,530   "block": 22,531   "row": 13012532  },533  {534   "path": "docs_nnx/guides/pytree.ipynb",535   "sha256": "2d895663391308464e5349d901ab8407c66825543da2b7d2eb323cff0c23dd24",536   "language": "Jupyter",537   "lines": 505,538   "truncated": false,539   "block": 22,540   "row": 13532541  },542  {543   "path": "docs_nnx/guides/randomness.ipynb",544   "sha256": "79bdc52d338aa6317cd7949f48349ba9f57ebd95925b36ed01bbce248b50eb38",545   "language": "Jupyter",546   "lines": 312,547   "truncated": false,548   "block": 22,549   "row": 13493550  },551  {552   "path": "docs_nnx/guides/surgery.ipynb",553   "sha256": "28f38931ac6d627bda89f3f18d1fadadd93a8317bbaaadc59cf83cb093fd233e",554   "language": "Jupyter",555   "lines": 223,556   "truncated": false,557   "block": 22,558   "row": 13207559  },560  {561   "path": "docs_nnx/guides/tiny_nnx.ipynb",562   "sha256": "f7603806f7504f890f65c5f8224c0d003fba94c68b37c6663e951ff674347d6e",563   "language": "Jupyter",564   "lines": 358,565   "truncated": false,566   "block": 22,567   "row": 13216568  },569  {570   "path": "docs_nnx/guides/transforms.ipynb",571   "sha256": "4fe7152a0e33b6dae3c937b9dc58bf3811b1aaa517a066fa21edea030dc4e4ec",572   "language": "Jupyter",573   "lines": 406,574   "truncated": false,575   "block": 22,576   "row": 13580577  },578  {579   "path": "docs_nnx/guides/view.ipynb",580   "sha256": "83bb894e706d2f9c886dd38659c3fd279765a810057145f43dee4ce55094796b",581   "language": "Jupyter",582   "lines": 338,583   "truncated": false,584   "block": 22,585   "row": 13469586  },587  {588   "path": "docs_nnx/hijax/hijax.ipynb",589   "sha256": "d621ee1285a74e8cc6b0093623e54ff74f6615b4339414b2e18328d8d1b9d693",590   "language": "Jupyter",591   "lines": 271,592   "truncated": false,593   "block": 22,594   "row": 12911595  },596  {597   "path": "docs_nnx/key_concepts.ipynb",598   "sha256": "a9598da3d58f7c28c297ee841e7d2aba47f4b51a6e6d529abd07f55919323c7d",599   "language": "Jupyter",600   "lines": 215,601   "truncated": false,602   "block": 22,603   "row": 13328604  },605  {606   "path": "docs_nnx/mnist_tutorial.ipynb",607   "sha256": "926ac2f48b9f95035aa6ee95cf3bd583226b4b0a2d4318e032c41e2a9cbe0c2f",608   "language": "Jupyter",609   "lines": 267,610   "truncated": false,611   "block": 22,612   "row": 13386613  },614  {615   "path": "docs_nnx/nnx_basics.ipynb",616   "sha256": "f2fb7a702adc4bab4fa258bfafaed419b8e4b45fe9a3eae3a4e8e13049f68207",617   "language": "Jupyter",618   "lines": 312,619   "truncated": false,620   "block": 22,621   "row": 13466622  },623  {624   "path": "examples/__init__.py",625   "sha256": "8642bbbe77864a965e977265c80b972defad8208b1d5d93a36c498c2141739d1",626   "language": "Python",627   "lines": 14,628   "truncated": false,629   "block": 22,630   "row": 27568631  },632  {633   "path": "examples/cloud/launch_gce.py",634   "sha256": "03687c4390390b51e76dcadd29fd833dfef75bb42b728d8f1671582adeddff8e",635   "language": "Python",636   "lines": 305,637   "truncated": false,638   "block": 23,639   "row": 9879640  },641  {642   "path": "examples/cloud/startup_script.sh",643   "sha256": "bb45eda8e3438a40568295109bfcbc5d2061a295e23266197c2c2d58f8efd3cd",644   "language": "Shell",645   "lines": 77,646   "truncated": false,647   "block": 23,648   "row": 18045649  },650  {651   "path": "examples/gemma/configs/benchmarks/gemma3_1b_grain.py",652   "sha256": "0700c88c836e4fd3cbbf622fe70ed4633e25071b6c21778693118052197a23f7",653   "language": "Python",654   "lines": 110,655   "truncated": false,656   "block": 23,657   "row": 4737658  },659  {660   "path": "examples/gemma/configs/benchmarks/gemma3_1b_tf.py",661   "sha256": "98c5fd5113bcde557f9bf66475127f6e86ec85b454d0b5d2e554c725f1b3068d",662   "language": "Python",663   "lines": 110,664   "truncated": false,665   "block": 23,666   "row": 4736667  },668  {669   "path": "examples/gemma/configs/default.py",670   "sha256": "e510fdb3caeddcfdc817ec09c938cd1db982ff221cd82e44710c0b3eec0d7de0",671   "language": "Python",672   "lines": 121,673   "truncated": false,674   "block": 23,675   "row": 5441676  },677  {678   "path": "examples/gemma/configs/gemma3_270m.py",679   "sha256": "dd5b341b074e7f3ff3eab157dc8930d37ce05276009b8516662e0323825f8df6",680   "language": "Python",681   "lines": 121,682   "truncated": false,683   "block": 23,684   "row": 5547685  },686  {687   "path": "examples/gemma/configs/gemma3_270m_sow.py",688   "sha256": "933a24a9810bfa61776e70de91938c8c7ace06ab851e89ff7b60e1ee42a14c8d",689   "language": "Python",690   "lines": 129,691   "truncated": false,692   "block": 23,693   "row": 5777694  },695  {696   "path": "examples/gemma/configs/gemma3_4b.py",697   "sha256": "6cb55f9ef4d9cf8bde27e50aa3df9724479af9fb26262130a64c52d5ec80f146",698   "language": "Python",699   "lines": 121,700   "truncated": false,701   "block": 23,702   "row": 5439703  },704  {705   "path": "examples/gemma/configs/small.py",706   "sha256": "0dca0204b91429af361fde5fb974be0b471bfbf96d3753c58bc4e9223680f9e4",707   "language": "Python",708   "lines": 143,709   "truncated": false,710   "block": 23,711   "row": 6347712  },713  {714   "path": "examples/gemma/configs/tiny.py",715   "sha256": "25e46eeb8e8503980ba4501c97be0dbc3166a5be9aceafa73ec4e07df8e1ee96",716   "language": "Python",717   "lines": 134,718   "truncated": false,719   "block": 23,720   "row": 5982721  },722  {723   "path": "examples/gemma/helpers.py",724   "sha256": "b733b2143729f9fa4a25d277f2814e6e720231ecf9dc22a7cce1a23156c393e9",725   "language": "Python",726   "lines": 93,727   "truncated": false,728   "block": 23,729   "row": 3644730  },731  {732   "path": "examples/gemma/helpers_test.py",733   "sha256": "9c8d85a11f539975e10664a1e55150252e91b14d7c0b8849d9c457437419e6d2",734   "language": "Python",735   "lines": 150,736   "truncated": false,737   "block": 23,738   "row": 5751739  },740  {741   "path": "examples/gemma/input_pipeline.py",742   "sha256": "aea610506c653965c9086bb702f0c009c36a24f9ac741fe82603d2b372a07b7d",743   "language": "Python",744   "lines": 40,745   "truncated": false,746   "block": 22,747   "row": 29485748  },749  {750   "path": "examples/gemma/input_pipeline_grain.py",751   "sha256": "f11390723fde87bc882cf0ae720baf391b0338e6db7afd59e26fa8272fb4b8d3",752   "language": "Python",753   "lines": 291,754   "truncated": false,755   "block": 23,756   "row": 9893757  },758  {759   "path": "examples/gemma/input_pipeline_test.py",760   "sha256": "edbfd4fb7bef60308a3108404b966eddd489e4c92a5518915bbafc8c0bcdd6c1",761   "language": "Python",762   "lines": 132,763   "truncated": false,764   "block": 23,765   "row": 5633766  },767  {768   "path": "examples/gemma/input_pipeline_tf.py",769   "sha256": "4487846e835c93c40a40d74543f3f7c847d24908fef0f0bfa0b7fd0149ef9409",770   "language": "Python",771   "lines": 427,772   "truncated": false,773   "block": 23,774   "row": 11725775  },776  {777   "path": "examples/gemma/layers.py",778   "sha256": "cc67d292898abdc09ae8bb85053d7a236bbe12de76809fc46411853064829458",779   "language": "Python",780   "lines": 59,781   "truncated": false,782   "block": 23,783   "row": 1196784  },785  {786   "path": "examples/gemma/layers_test.py",787   "sha256": "2a52625442f7985e6373de983d91ce9d519bd419a87006f5a0c325e5307d9c09",788   "language": "Python",789   "lines": 35,790   "truncated": false,791   "block": 22,792   "row": 29416793  },794  {795   "path": "examples/gemma/main.py",796   "sha256": "07c92fcb79413901ff5d653ee5c3134767182c38a1db5b8df1666c2d43700f67",797   "language": "Python",798   "lines": 88,799   "truncated": false,800   "block": 23,801   "row": 3111802  },803  {804   "path": "examples/gemma/modules.py",805   "sha256": "f46be9406b59f777935bf9750fa2c79590d63ae1feaf5ff777646cc24e5f75e3",806   "language": "Python",807   "lines": 544,808   "truncated": false,809   "block": 23,810   "row": 12593811  },812  {813   "path": "examples/gemma/modules_test.py",814   "sha256": "8bc853410ac33cfcfea0641d3e1b3781986d44b01b82e29299ae1770d17038f9",815   "language": "Python",816   "lines": 454,817   "truncated": false,818   "block": 23,819   "row": 11527820  },821  {822   "path": "examples/gemma/params.py",823   "sha256": "e3af47abd9d13f1ca1b2246be0b2322859dd5412bf5b8f3f3b9ac70a176123ed",824   "language": "Python",825   "lines": 92,826   "truncated": false,827   "block": 23,828   "row": 3048829  },830  {831   "path": "examples/gemma/positional_embeddings.py",832   "sha256": "58ae97156969f8f417b437d1e5c6afd8fc7fdc796bec097dff0ab7d899dcf162",833   "language": "Python",834   "lines": 71,835   "truncated": false,836   "block": 23,837   "row": 2389838  },839  {840   "path": "examples/gemma/positional_embeddings_test.py",841   "sha256": "9a33640973c1a2596d139a41235d908a6e13b1173449ee4fe42f79244a22e670",842   "language": "Python",843   "lines": 79,844   "truncated": false,845   "block": 23,846   "row": 2841847  },848  {849   "path": "examples/gemma/sampler.py",850   "sha256": "7b0c532bc951e5551c4a86c27ebfafe87d3d530b91b1c49c0db1f2b1bebe9295",851   "language": "Python",852   "lines": 550,853   "truncated": false,854   "block": 23,855   "row": 12921856  },857  {858   "path": "examples/gemma/sampler_test.py",859   "sha256": "9ac24aa62c7448b3f78dd78502e000ee981c11bee40749c2d18dcabe91c02917",860   "language": "Python",861   "lines": 589,862   "truncated": false,863   "block": 23,864   "row": 13154865  },866  {867   "path": "examples/gemma/sow_lib.py",868   "sha256": "f9a8613fbf7a3ec68d664e9ba3184a94c5dc414102d3f1e503fd7cda79599a4b",869   "language": "Python",870   "lines": 259,871   "truncated": false,872   "block": 23,873   "row": 9676874  },875  {876   "path": "examples/gemma/tokenizer.py",877   "sha256": "bc769299e902f9c9dbd4edf51c8e8f46571f477a7c733870b60c4c7a09431f28",878   "language": "Python",879   "lines": 216,880   "truncated": false,881   "block": 23,882   "row": 8447883  },884  {885   "path": "examples/gemma/train.py",886   "sha256": "2a4bd9038f0fc91671ece85820486852fac75a01e538d3b88f379f3dd171c627",887   "language": "Python",888   "lines": 714,889   "truncated": false,890   "block": 23,891   "row": 13651892  },893  {894   "path": "examples/gemma/train_cfg.py",895   "sha256": "87504fed6ebb16fc726e5b4f4a9d156d4d6d1581f1daedf4ea73ccf16db2d4a4",896   "language": "Python",897   "lines": 127,898   "truncated": false,899   "block": 23,900   "row": 5356901  },902  {903   "path": "examples/gemma/transformer.py",904   "sha256": "6c17379eb1fc546ea68f752c3a5e8805314ee64ac47cd5f7500386704678efa6",905   "language": "Python",906   "lines": 252,907   "truncated": false,908   "block": 23,909   "row": 8804910  },911  {912   "path": "examples/gemma/transformer_cfg.py",913   "sha256": "0101dc6ff36513d5f3d25265f265f51e6022a3f7f812d49854e4887e0ccc71ec",914   "language": "Python",915   "lines": 605,916   "truncated": false,917   "block": 23,918   "row": 12955919  },920  {921   "path": "examples/gemma/transformer_test.py",922   "sha256": "7c73c2ae40372bd1e48bd00394241c7b5a22b7297e05115bb75a4217248be01b",923   "language": "Python",924   "lines": 591,925   "truncated": false,926   "block": 23,927   "row": 13007928  },929  {930   "path": "examples/imagenet/configs/default.py",931   "sha256": "7f421cb274c32af999ba46187db79d005e602388f74638b57301b35fe09f270c",932   "language": "Python",933   "lines": 60,934   "truncated": false,935   "block": 23,936   "row": 706937  },938  {939   "path": "examples/imagenet/configs/fake_data_benchmark.py",940   "sha256": "cc81746e5674ec6d3e7a737ebef1a1b03acfe2fc09cd6bfb0c3c764086490a41",941   "language": "Python",942   "lines": 34,943   "truncated": false,944   "block": 22,945   "row": 29092946  },947  {948   "path": "examples/imagenet/configs/tpu.py",949   "sha256": "655e666b7754a16f391f6d7bec74761249a1729845708814ada5fee0f5fdb016",950   "language": "Python",951   "lines": 33,952   "truncated": false,953   "block": 22,954   "row": 29040955  },956  {957   "path": "examples/imagenet/configs/v100_x8.py",958   "sha256": "4c5f3ebf212a692792daf1c6d972ecf71fbfcea63e8e3b157e086633f6c26d28",959   "language": "Python",960   "lines": 32,961   "truncated": false,962   "block": 22,963   "row": 28968964  },965  {966   "path": "examples/imagenet/configs/v100_x8_mixed_precision.py",967   "sha256": "99bb0402c112f483a9072125f3f168f208e031259d51b6b47afd9738bd9a863b",968   "language": "Python",969   "lines": 33,970   "truncated": false,971   "block": 22,972   "row": 29042973  },974  {975   "path": "examples/imagenet/imagenet.ipynb",976   "sha256": "13f996a4542f20419ff8aa53284c4c2754876709e528d77407a2bbe1a9bb6848",977   "language": "Jupyter",978   "lines": 350,979   "truncated": false,980   "block": 22,981   "row": 13311982  },983  {984   "path": "examples/imagenet/imagenet_benchmark.py",985   "sha256": "69ec9632576e9da4f4d691d04c12f0f454108a3470eac9787eade4e46be45696",986   "language": "Python",987   "lines": 102,988   "truncated": false,989   "block": 23,990   "row": 3919991  },992  {993   "path": "examples/imagenet/imagenet_fake_data_benchmark.py",994   "sha256": "0ecc532c634e097cd6c53860939fe33505e34ef64142d7744903fa55c7b629c3",995   "language": "Python",996   "lines": 65,997   "truncated": false,998   "block": 23,999   "row": 18221000  },1001  {1002   "path": "examples/imagenet/input_pipeline.py",1003   "sha256": "566659dd4973f2879a3856b379dd21bfeba4bd7411f06a765484de185ee64df4",1004   "language": "Python",1005   "lines": 254,1006   "truncated": false,1007   "block": 23,1008   "row": 92341009  },1010  {1011   "path": "examples/imagenet/main.py",1012   "sha256": "0ad0328f396d9e50b8f1bf468fe2cc19d1b453252428fc7a2890aeb603589e63",1013   "language": "Python",1014   "lines": 69,1015   "truncated": false,1016   "block": 23,1017   "row": 17021018  },1019  {1020   "path": "examples/imagenet/models.py",1021   "sha256": "d3784cfa4d7dda64ba532a4793ff4112cd8bae584a9f00477c294e57e0b59e85",1022   "language": "Python",1023   "lines": 163,1024   "truncated": false,1025   "block": 23,1026   "row": 54091027  },1028  {1029   "path": "examples/imagenet/models_test.py",1030   "sha256": "3b636613e744fe59a1e6c135b108ba3199f91fc957dcf33f4722749db098b0bf",1031   "language": "Python",1032   "lines": 57,1033   "truncated": false,1034   "block": 23,1035   "row": 12591036  },1037  {1038   "path": "examples/imagenet/train.py",1039   "sha256": "aa4a8b3cf87f4467a5a3d3b785532274acf33766008a478cfe931240065f2593",1040   "language": "Python",1041   "lines": 428,1042   "truncated": false,1043   "block": 23,1044   "row": 114011045  },1046  {1047   "path": "examples/imagenet/train_test.py",1048   "sha256": "497d7ad8e037d85ef51a7983c2e7e4d3b6e02e29af2b2efd1bb24be4f5b59bc3",1049   "language": "Python",1050   "lines": 90,1051   "truncated": false,1052   "block": 23,1053   "row": 34361054  },1055  {1056   "path": "examples/linen_design_test/attention_simple.py",1057   "sha256": "c33b7432829d98cb62ebbb65df29505dd6737cb7f56fcdbff8202a09e593ac70",1058   "language": "Python",1059   "lines": 219,1060   "truncated": false,1061   "block": 23,1062   "row": 76361063  },1064  {1065   "path": "examples/linen_design_test/autoencoder.py",1066   "sha256": "d7f73e7eb3e94b8ecd7d847b88624f17007f3f5f6d11afdee50dea8f1c08ee36",1067   "language": "Python",1068   "lines": 100,1069   "truncated": false,1070   "block": 23,1071   "row": 36621072  },1073  {1074   "path": "examples/linen_design_test/dense.py",1075   "sha256": "af5b90a861b8ae9ed66c8873685b9804b7842730d396eaa64a774be2933005ec",1076   "language": "Python",1077   "lines": 40,1078   "truncated": false,1079   "block": 22,1080   "row": 294571081  },1082  {1083   "path": "examples/linen_design_test/linear_regression.py",1084   "sha256": "7fd126f949b58d0a296938840852c6ddebc8daa53322db88450269f734a10af4",1085   "language": "Python",1086   "lines": 51,1087   "truncated": false,1088   "block": 22,1089   "row": 295591090  },1091  {1092   "path": "examples/linen_design_test/mlp_explicit.py",1093   "sha256": "44861e9be23c4cde610ea3e5a69d6a493919430df39a3e0167fbb2fb767e7bea",1094   "language": "Python",1095   "lines": 67,1096   "truncated": false,1097   "block": 23,1098   "row": 20601099  },1100  {1101   "path": "examples/linen_design_test/mlp_inline.py",1102   "sha256": "c2ebb4dbc474df57124a55c8eb91fef29206a8c730dc42ee3c9105381912fcb5",1103   "language": "Python",1104   "lines": 52,1105   "truncated": false,1106   "block": 23,1107   "row": 11511108  },1109  {1110   "path": "examples/linen_design_test/mlp_lazy.py",1111   "sha256": "b4e5dd8cf36a53fcdd44eb436deb21589151a829874fc64cd104328e5c5a3299",1112   "language": "Python",1113   "lines": 53,1114   "truncated": false,1115   "block": 23,1116   "row": 11561117  },1118  {1119   "path": "examples/lm1b/configs/default.py",1120   "sha256": "ca14b21936e56cf238ecf0961b974cca771e85a105f7a13f4720b0b8c8fb9905",1121   "language": "Python",1122   "lines": 149,1123   "truncated": false,1124   "block": 23,1125   "row": 59421126  },1127  {1128   "path": "examples/lm1b/input_pipeline.py",1129   "sha256": "51d3dd3c09a20f6b85c69c2641167d7a87af579478cef5c6a9e10759faeaf5c2",1130   "language": "Python",1131   "lines": 386,1132   "truncated": false,1133   "block": 23,1134   "row": 113571135  },1136  {1137   "path": "examples/lm1b/input_pipeline_test.py",1138   "sha256": "e722190ba7444c76bf22c6ed54038e288e504e58818d53d16a735df81fb1df50",1139   "language": "Python",1140   "lines": 104,1141   "truncated": false,1142   "block": 23,1143   "row": 41491144  },1145  {1146   "path": "examples/lm1b/main.py",1147   "sha256": "7a29740deae7c0fde8aa1c1ed0ba08fbed55f523319bb6e099558144251aa5b1",1148   "language": "Python",1149   "lines": 69,1150   "truncated": false,1151   "block": 23,1152   "row": 18231153  },1154  {1155   "path": "examples/lm1b/models.py",1156   "sha256": "938782fe5d0387e86f0f00e138be2166016e6af1546c12d3b79ba7b56c7d78b8",1157   "language": "Python",1158   "lines": 426,1159   "truncated": false,1160   "block": 23,1161   "row": 116051162  },1163  {1164   "path": "examples/lm1b/temperature_sampler.py",1165   "sha256": "16d145a7c59e92c254dcd7001231e0bbb830722bf903f58ed478556f8ab26709",1166   "language": "Python",1167   "lines": 127,1168   "truncated": false,1169   "block": 23,1170   "row": 60821171  },1172  {1173   "path": "examples/lm1b/temperature_sampler_test.py",1174   "sha256": "c406e56b3a9cdb5ea44ad705fb412aabc07a27fecbd93a589841c166dd612bc6",1175   "language": "Python",1176   "lines": 48,1177   "truncated": false,1178   "block": 23,1179   "row": 1741180  },1181  {1182   "path": "examples/lm1b/tokenizer.py",1183   "sha256": "4799ef896c03ca729e411132857b2535422b9f7ef18aafb7f407945c5ba7db62",1184   "language": "Python",1185   "lines": 169,1186   "truncated": false,1187   "block": 23,1188   "row": 67801189  },1190  {1191   "path": "examples/lm1b/train.py",1192   "sha256": "2d61517f93ca6335c321c206e7cfc8be7b38fe68627e1550744bdc0527531eae",1193   "language": "Python",1194   "lines": 606,1195   "truncated": false,1196   "block": 23,1197   "row": 131571198  },1199  {1200   "path": "examples/lm1b/train_test.py",

Showing the first 1,200 of 3666 lines. Download the file for the rest.