The Wayback Machine - https://web.archive.org/web/20230215045408/https://github.com/google/flax/pulls
Skip to content

Pull requests: google/flax

New pull request New
52 Open 1,615 Closed
52 Open 1,615 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Enhance Orbax checkpointing documentation
#2870 opened Feb 14, 2023 by cpgaffney1 Loading…
1
Improve notebook handling
pull ready
#2860 opened Feb 9, 2023 by cgarciae Loading…
@cgarciae
2
feat(misc): create pyproject.toml
pull ready
#2848 opened Feb 6, 2023 by SauravMaheshkar Loading…
5
Improve Dropout guide
#2846 opened Feb 3, 2023 by cgarciae • Draft
2
Fix lm1b sampler
#2842 opened Feb 2, 2023 by cgarciae Loading…
1
1
Improve nn.scan docs
#2839 opened Feb 2, 2023 by cgarciae Loading…
1
3
Add Google Research Flax examples to docs
#2825 opened Jan 30, 2023 by 8bitmp3 • Draft
changed init and apply functions to return regular dicts instead of FrozenDicts
pull ready
#2822 opened Jan 28, 2023 by chiamp Loading…
1
5
A small doc and one fixe to Imagenet benchmark
#2812 opened Jan 19, 2023 by nouiz Loading…
2
Implement BoxedParam as flax.core.AxisMetadata.
#2789 opened Jan 12, 2023 by copybara-service bot Loading…
1
Added initializers doc
#2776 opened Jan 7, 2023 by chiamp Loading…
1
@chiamp
12
Register Jax key paths for FrozenDict
#2771 opened Jan 4, 2023 by JesseFarebro Loading…
2 of 4 tasks
[WIP] Add Bidirectional
#2770 opened Jan 4, 2023 by cgarciae • Draft
@cgarciae
forced post init to call super post init
#2761 opened Dec 30, 2022 by chiamp Loading…
@chiamp
3
Bind submodules using __getattribute__
#2748 opened Dec 20, 2022 by cgarciae Loading…
1
@cgarciae
2
added deep mutation contextmanager for dataclasses
#2735 opened Dec 17, 2022 by chiamp Loading…
@chiamp
5
Add support for PyTrees with None leaves. Add additional unit tests.
#2719 opened Dec 12, 2022 by copybara-service bot Loading…
Add Stateful Modules guide
#2689 opened Dec 5, 2022 by cgarciae Loading…
@cgarciae
8
Updated flax docs front page
#2671 opened Nov 29, 2022 by chiamp Loading…
@chiamp
8
Created colab_snippets.ipynb
#2669 opened Nov 29, 2022 by chiamp • Draft
@chiamp
4
Add Flax VAE tutorial
#2650 opened Nov 23, 2022 by 8bitmp3 • Draft
@8bitmp3
1
Testing out pyink formatting.
Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2608 opened Nov 12, 2022 by levskaya Loading…
@levskaya
22
Added check_shapes functionality when reading from bytes
#2607 opened Nov 12, 2022 by chiamp • Draft
@chiamp
1
RNN Module
#2604 opened Nov 10, 2022 by cgarciae Loading…
1
Adds flax.training.gfile_utils for storing logs on GCS.
Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2603 opened Nov 10, 2022 by andsteing Loading…
@andsteing
1
ProTip! Filter pull requests by the default branch with base:main.