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

178 Open 747 Closed
178 Open 747 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

Flax NNX and Orbax Checkpointing require hacking to work together
#4383 opened Nov 15, 2024 by hdrwilkinson
2
JAX for the impatient was deleted, but it is still referenced twice
#4382 opened Nov 15, 2024 by ArthurConmy
1
7x7 nnx.Conv using float32 parameter dtype overflows(?) to nan when sharded
#4375 opened Nov 12, 2024 by joshhansen
Initialization of flax.linen model parameters depends on member variable names
#4367 opened Nov 10, 2024 by bluesunb
Scaling issue on CPU (X86 and arm) while using models from huffing face.
#4357 opened Nov 5, 2024 by choudhary-devang
nnx.vmap example from documentation raise an index error
#4355 opened Nov 4, 2024 by jhn-nt
5
Slow training step occasionally due to slow graph flatten Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4336 opened Oct 25, 2024 by kriscao-cohere
@cgarciae
3
nnx.pmap documentation seems to be for vmap Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4330 opened Oct 24, 2024 by hrbigelow
@cgarciae
2
Bug - Cannot access attribute "replace" for class "c123"
#4312 opened Oct 20, 2024 by MRiabov
5
Can't save checkpoint with orbax when using zero-size parameters
#4309 opened Oct 18, 2024 by dionhaefner
@IvyZX
3
NNX Conv Layer Input Tuple Error Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
#4295 opened Oct 14, 2024 by riverliway
@8bitmp3
1
NNX documentation missing pooling operations
#4271 opened Oct 8, 2024 by tux-type
@cgarciae @8bitmp3
3
Add nnx.shard_map Priority: P0 (urgent) Response within 1 business day. Resolution as soon as possible. (Assignee required)
#4261 opened Oct 7, 2024 by cgarciae
@cgarciae
1
Pickling a trained model (NNX)
#4247 opened Oct 3, 2024 by eugene
7
How to use max_pool with xla_gpu_deterministic_ops?
#4233 opened Sep 28, 2024 by blackblitz
Add NNX support for legacy jax.random.PRNGKey()
#4231 opened Sep 26, 2024 by cisprague
1
nnx.jit(aux_fn) is slower than directly using nnx.jit(model.__call__)
#4218 opened Sep 23, 2024 by JunhongXu
3
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
2
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
ProTip! Adding no:label will show everything without a label.