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

138 Open 717 Closed
138 Open 717 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
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

typo in nnx_basics.md
#4046 opened Jul 1, 2024 by notnot
Significant performance difference of NNX relative to equinox
#4045 opened Jun 30, 2024 by jlperla
3
lstm error
#4032 opened Jun 26, 2024 by layssi
1
Suboptimal default initialization of q/k/v projections in nn.MultiHeadDotProductAttention Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#4027 opened Jun 25, 2024 by MasterSkepticista
2
flax nn.tabulate Incorrectly Reports FLOPs and VJP FLOPs
#4023 opened Jun 24, 2024 by Surya-77
2
serialization.from_state_dict does not restore to jax.Arrays
#3999 opened Jun 15, 2024 by PhilipVinc
1
Best practice of dealing with sporadic FrozenDict conversions?
#3994 opened Jun 13, 2024 by TimSchneider42
2
New python venv fails to pip install the flax mnist example requirements
#3990 opened Jun 12, 2024 by sycamoreoak
3
Using vmap on the loss function change results
#3973 opened Jun 6, 2024 by gduflo
feature_request: support for tabulate/summary in the NNX API Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#3962 opened Jun 4, 2024 by AshishKumar4
1
imagenet example needs a refreshed requirements.txt
#3950 opened May 29, 2024 by sycamoreoak
1
static_argnums argument to flax.linen.remat not working as expected Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3946 opened May 28, 2024 by dionhaefner
@cgarciae
2
Problem in usage of multi-gpu (Four RTX 4090) for FLAX Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#3931 opened May 22, 2024 by psj1866
6
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
ProTip! Mix and match filters to narrow down what you’re looking for.