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

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

MNIST NNX Tutorial broken links
#4394 opened Nov 22, 2024 by stergiosba
1
5
Issue with Optimizer Update in A2C Network with Optax Body:
#4391 opened Nov 20, 2024 by Tomato-toast
6
ResNet-18 Initialization
#4388 opened Nov 19, 2024 by ZedongPeng
Flax NNX and Orbax Checkpointing require hacking to work together
#4383 opened Nov 15, 2024 by hdrwilkinson
3
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
1
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
5
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
10
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
ProTip! Adding no:label will show everything without a label.