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.
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
Documentation links 404
Priority: P1 - soon
Response within 5 business days. Resolution within 30 days. (Assignee required)
#3917
opened May 9, 2024 by
namiller
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
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
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
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
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
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
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
Remove Best effort response and resolution. We have no plan to work on this at the moment.
tree_map
deprecation filter after Flax upgrades minimum Python version to 3.10
Priority: P2 - no schedule
#3842
opened Apr 10, 2024 by
chiamp
In Best effort response and resolution. We have no plan to work on this at the moment.
MultiHeadAttention
, let num_heads=1
by default
Priority: P2 - no schedule
#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 Best effort response and resolution. We have no plan to work on this at the moment.
BatchRenorm
layer to linen.normalization
Priority: P2 - no schedule
#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
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
Error when calling Response within 5 business days. Resolution within 30 days. (Assignee required)
Module.tabulate
on normalization wrappers like WeightNorm
and SpectralNorm
Priority: P1 - soon
#3779
opened Mar 21, 2024 by
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
*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
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
Error occurs in Best effort response and resolution. We have no plan to work on this at the moment.
nn.vmap
while variable_axes
is a nested dict
Priority: P2 - no schedule
#3751
opened Mar 12, 2024 by
egg5154
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
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
Make redundant Best effort response and resolution. We have no plan to work on this at the moment.
features
argument optional for recurrent cells
Priority: P2 - no schedule
#3717
opened Feb 25, 2024 by
carlosgmartin
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
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
Previous Next
ProTip!
What’s not been updated in a month: updated:<2024-04-21.