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

126 Open 704 Closed
126 Open 704 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

nnx.graph.split infinite recursion when used in a thread
#3899 opened May 3, 2024 by NeilGirdhar
@cgarciae
1
with_partitioning has surprising behavior with MultiHeadAttention and DenseGeneral Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3892 opened Apr 30, 2024 by jameslyon
1
Tutorial request Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3890 opened Apr 30, 2024 by Wasim04
2
nn.remat_scan doesn't work with nn.with_partitioning Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3870 opened Apr 19, 2024 by jameslyon
1
Problem while using checkpoints.restore_checkpoint with gradio Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3860 opened Apr 16, 2024 by GCP20
1
Add ability to easily change documentation version Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3858 opened Apr 15, 2024 by emergenz
@chiamp
Improve SEO for docs pages Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3854 opened Apr 12, 2024 by carlosgmartin
1
@chiamp
2
Unpickled modules with constructor arguments cannot be initialized Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3846 opened Apr 11, 2024 by eseraygun
@cgarciae
Remove tree_map deprecation filter after Flax upgrades minimum Python version to 3.10 Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3842 opened Apr 10, 2024 by chiamp
1
@chiamp
In MultiHeadAttention, let num_heads=1 by default Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3833 opened Apr 8, 2024 by carlosgmartin
GroupedConv distributed training failure Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3831 opened Apr 8, 2024 by sali1997s
Add BatchRenorm layer to linen.normalization Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3822 opened Apr 3, 2024 by danielpalen
Using variable declared at a broader scope in a function is bad form Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3800 opened Mar 30, 2024 by afairley
1
For some reason these imports are elided on read the docs Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3799 opened Mar 30, 2024 by afairley
1
Error when calling Module.tabulate on normalization wrappers like WeightNorm and SpectralNorm Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3779 opened Mar 21, 2024 by chiamp
@cgarciae @chiamp
More memory consume compared with Pytorch Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3766 opened Mar 18, 2024 by Sun-Xiaohui
1
*Module Parameters* section of docs is outdated. Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3761 opened Mar 14, 2024 by PaulScemama
@chiamp
4
flax.linen.module.init still fails under dynamic type checking for nested modules Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3756 opened Mar 13, 2024 by evangelos-ch
@cgarciae
Error occurs in nn.vmap while variable_axes is a nested dict Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3751 opened Mar 12, 2024 by egg5154
1
Neural Net Training is bottlenecked by maxed out CPU Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3750 opened Mar 12, 2024 by josephdepaoloboisvert
Wrong parameter names when nesting Modules within flax transformations Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3747 opened Mar 11, 2024 by PhilipVinc
@cgarciae
3
modifying params of flax.linen. Module model Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3730 opened Feb 29, 2024 by lopa23
1
Make redundant features argument optional for recurrent cells Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3717 opened Feb 25, 2024 by carlosgmartin
4
The Error in FLOP Computation of Model Tabulate Function Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3716 opened Feb 24, 2024 by GoktugGuvercin
@cgarciae
1
Feature request: Add ConvGRUCell Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3713 opened Feb 24, 2024 by carlosgmartin
ProTip! Add no:assignee to see everything that’s not assigned.