Skip to content

Issues: google/flax

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

75 Open 567 Closed
75 Open 567 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
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Load checkpoint on CPU backend throws error
#2866 opened Feb 13, 2023 by cap-jmk
@IvyZX
1
Unexpected ScopeParamNotFoundError when sharing submodules Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2864 opened Feb 10, 2023 by kvablack
@cgarciae
1
Flax doesn't work on google colab Priority: P0 (urgent) An issue of the highest priority. We are addressing this urgently. (Assignee required)
#2861 opened Feb 9, 2023 by necludov
8
Feature request: FlashAttention
#2858 opened Feb 9, 2023 by OhadRubin
Incorporate RMSNorm Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2849 opened Feb 6, 2023 by MechCoder
1
AssertionError: sync_global_devices name mismatch ('starting_save_checkpoint') Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2834 opened Feb 1, 2023 by jenkspt
@IvyZX
3
switching to _src structure Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2814 opened Jan 23, 2023 by chiamp
GSoC'23
#2813 opened Jan 22, 2023 by shivance
1
Feature Request: Example Documentation Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2782 opened Jan 9, 2023 by marcvanzee
4
Module.init does not contain KV-pair of modules without parameters Priority: P3 - no schedule We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#2777 opened Jan 8, 2023 by epignatelli
4
Rename ConvLSTM Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2765 opened Jan 2, 2023 by cgarciae
@cgarciae
Improve pjit guide Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2755 opened Dec 23, 2022 by IvyZX
@IvyZX
Initialization of Submodules Lifted with flax.nn.scan
#2754 opened Dec 22, 2022 by daskol
1
@cgarciae
6
[RTD] Python syntax highlighting doesn't always work Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2753 opened Dec 22, 2022 by marcvanzee
@8bitmp3
2
Make a guide for scan over layers Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2752 opened Dec 22, 2022 by IvyZX
@IvyZX
1
Common Initializers Does Not Work with Bias Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2749 opened Dec 21, 2022 by daskol
1
@chiamp
6
Properly bind all submodules Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2744 opened Dec 20, 2022 by cgarciae
1
Wrap __post_init__ to check or force that Module.__post_init__ is called Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2733 opened Dec 17, 2022 by levskaya
@levskaya
2
Use enterprise test runners for our CI once they are working again Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2731 opened Dec 16, 2022 by marcvanzee
@marcvanzee
1
Support loading and fine-tuning TensorFlow models Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2721 opened Dec 13, 2022 by cgarciae
@cgarciae
1
0.6.3 introduced a circular dependency with orbax Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2707 opened Dec 10, 2022 by mattsoulanille
@IvyZX
13
Add glossary page in flax docs Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2681 opened Dec 1, 2022 by chiamp
@chiamp
3
Enable internal type checking for tests Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2679 opened Dec 1, 2022 by marcvanzee
Rename flax.linen to flax.nn? Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#2677 opened Dec 1, 2022 by PhilipVinc
@marcvanzee
3
Issue with mkdocstrings since recent changes on struct.dataclass Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#2648 opened Nov 22, 2022 by felixchalumeau
@cgarciae
5
ProTip! Mix and match filters to narrow down what you’re looking for.