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

163 Open 731 Closed
163 Open 731 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

nnx.Swish, jax.swish,... change the input shape
#4214 opened Sep 21, 2024 by leson207
2
Depthwise convolution in flax.nnx.Conv is significantly slower than PyTorch and TensorFlow
#4207 opened Sep 18, 2024 by YushaArif99
nnx.vmap use the same random key rngs inside nnx.Module across vectorization.
#4195 opened Sep 15, 2024 by maxencefaldor
1
TODO: add notes link in flax_basics guide
#4193 opened Sep 13, 2024 by garymm
flax counterpart for torch.nn.Conv1d
#4188 opened Sep 10, 2024 by Liyang90
MultiHeadAttention documentation missing descriptions of None values for optional arguments
#4182 opened Sep 9, 2024 by carlosgmartin
Contributing to Flax Community Tutorials
#4181 opened Sep 9, 2024 by yilunyu
1
CHANGELOG has not been updated since 0.8.2
#4179 opened Sep 7, 2024 by enolan
How to freeze parameters with nnx and optax?
#4167 opened Sep 1, 2024 by maxencefaldor
6
ToLinen is not hashable (Linen modules are)
#4156 opened Aug 29, 2024 by PhilipVinc
2
Support for optax lbfgs and related optimizers with NNX Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4144 opened Aug 26, 2024 by jlperla
1
Experimental-pytree flag causes crash
#4142 opened Aug 24, 2024 by NeilGirdhar
2
nn.WeightNorm needs a special scale_init to match PyTorch weight_norm
#4138 opened Aug 21, 2024 by DBraun
1
Add IndRNN
#4133 opened Aug 20, 2024 by carlosgmartin
1
Flax dataclasses do not support new-style PRNGKeyArray in serialisation Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4130 opened Aug 19, 2024 by PhilipVinc
SPMD for initializing model using nnx.jit Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4129 opened Aug 18, 2024 by mmorinag127
1
Clarification for LSTMCell Documentation Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
#4124 opened Aug 12, 2024 by corentinlger
2
MNIST tutorial broken for Colab TPU
#4122 opened Aug 7, 2024 by rcrowe-google
bias and kernel params are put on different gpu devices
#4116 opened Aug 6, 2024 by YunxiTang
1
DynamicScale behaves unexpected when computing per-sample gradients with vmap.
#4114 opened Aug 2, 2024 by hlzl
[FLIP] JAX-style NNX Transforms
#4107 opened Jul 26, 2024 by cgarciae
Performance penalty of typecast in nn.Embed.__call__
#4100 opened Jul 23, 2024 by MasterSkepticista
lowering / cost analysis of @nnx.jit functions Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4094 opened Jul 19, 2024 by cgarciae
1
Truncated Normal initializer doesn't match PyTorch
#4091 opened Jul 18, 2024 by DBraun
2
NNXWrapper
#4088 opened Jul 18, 2024 by PhilipVinc
2
ProTip! Add no:assignee to see everything that’s not assigned.