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

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

Why doesn't nnx.fori_loop work here?
#4436 opened Dec 13, 2024 by cgarciae
Support for exporting models in the ONNX format
#4430 opened Dec 12, 2024 by Artoriuz
2
NNX WeightNorm Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment. Status: pull requests welcome We agree with the direction proposed, feel free to give it a shot and file a pull request
#4426 opened Dec 10, 2024 by rbflx
1
Issues checkpointing optimizer state using Optax, nnx.Optimizer, and Orbax
#4423 opened Dec 9, 2024 by SandSnip3r
7
[FR] Save/load of model parameters
#4422 opened Dec 8, 2024 by PhilipVinc
2
Clarification on sharding strategy to combine multiple training steps together via nnx.scan
#4417 opened Dec 4, 2024 by Teculos
2
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
1
@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
ProTip! Adding no:label will show everything without a label.