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

99 Open 672 Closed
99 Open 672 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

Quick start badges point to 404
#3589 opened Dec 24, 2023 by MartinuzziFrancesco
Swapping selected layers with a different layer.
#3585 opened Dec 21, 2023 by aneesh-aparajit
1
Metadata in nnx.param_field is not passed to the Param class
#3577 opened Dec 20, 2023 by frazane
flax is significantly slower than pytorch
#3551 opened Dec 13, 2023 by alerem18
Bad interaction between nnx.Rngs and custom derivatives
#3528 opened Dec 4, 2023 by NeilGirdhar
@cgarciae
6
How to optimize multi model’s parameter in one optimizer ?
#3524 opened Dec 1, 2023 by GCP20
1
Add a mask argument to RNN.__call__
#3507 opened Nov 24, 2023 by carlosgmartin
1
Full Module RNG Scope
#3506 opened Nov 24, 2023 by joeryjoery
2
Multiple initializations. Is this a bug? 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
#3499 opened Nov 22, 2023 by henrypinkard
@cgarciae
5
Incorrect flop count estimate when using flax.linen.remat_scan
#3492 opened Nov 17, 2023 by transformer-vq
1
Create .ipynb notebooks for guides
#3457 opened Nov 2, 2023 by chiamp
3 tasks
@8bitmp3
1
Possibly Erroneous FLOPS Calculation
#3392 opened Oct 4, 2023 by foainla
2
Imagenet example does not work with newer flax versions
#3364 opened Sep 25, 2023 by gkroiz
1
32
Unexpected segmentation fault on GPU inside flax.linen.scan 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
#3362 opened Sep 24, 2023 by Kajiih
2
Show sharding information in Module.tabulate Priority: P2 - eventual This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#3325 opened Sep 12, 2023 by cgarciae
@cgarciae
Nested nn.vmap of nn.Module with nn.BatchNorm causes unexpected ValueError
#3319 opened Sep 11, 2023 by pseifner
AttributeError: module 'flax' has no attribute 'nn'
#3296 opened Aug 27, 2023 by AliAlhasan2
2
Regular dict representation not indented
#3280 opened Aug 15, 2023 by chiamp
3
initialization with parameters depends on input
#3276 opened Aug 11, 2023 by Roger-luo
@chiamp
1
exception flax.errors.InvalidRngError(msg)[source]
#3257 opened Aug 3, 2023 by haegray-curp
@chiamp
2
API to build non-sequential models 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
#3244 opened Jul 31, 2023 by epignatelli
2
TPU Colab fails with AttributeError: module 'jax.tree_util' has no attribute 'register_pytree_with_keys_class' Priority: P1 - soon Assignee is working on this now, among other tasks. (Assignee required)
#3239 opened Jul 31, 2023 by andsteing
@andsteing
1
Test failures: "RuntimeError: Attempted to set 4 devices, but 1 CPUs already available: ensure that set_n_cpu_devices is executed before any JAX operation."
#3232 opened Jul 27, 2023 by samuela
@chiamp
1
Inconsistency in keep argument for save_checkpoint and save_checkpoint_multiprocess
#3192 opened Jul 12, 2023 by priyakasimbeg
1
AttributeError: module 'jax.config' has no attribute 'define_bool_state' when running big_vision on tpu-vm-base
#3180 opened Jul 5, 2023 by yuyangshu
3
ProTip! Exclude everything labeled bug with -label:bug.