Skip to content

Pull requests: google/flax

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
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

[nnx] keras integration
#4396 opened Nov 23, 2024 by cgarciae Loading…
Missed MNIST Tutorial Added
#4395 opened Nov 22, 2024 by selamw1 Loading…
Automated Code Change
#4393 opened Nov 21, 2024 by copybara-service bot Loading…
Update async_checkpointer.py reference
#4385 opened Nov 16, 2024 by emmanuel-ferdman Loading…
1 of 4 tasks
[nnx] add performance guide notebook
#4384 opened Nov 15, 2024 by cgarciae Loading…
Add flax.nnx.value_and_grad docstring
#4372 opened Nov 11, 2024 by 8bitmp3 Loading…
Update flax.nnx.jit docstring
#4369 opened Nov 11, 2024 by 8bitmp3 Loading…
Update flax.nnx.grad docstring
#4368 opened Nov 11, 2024 by 8bitmp3 Loading…
Fix multiple links to Orbax documentation
#4364 opened Nov 8, 2024 by Matt-Hurd Loading…
4 tasks done
[nnx] add unsafe_graph_cache decorator
#4356 opened Nov 4, 2024 by cgarciae Loading…
[nnx] Optimizer uses variables
#4347 opened Oct 30, 2024 by cgarciae Loading…
Upgrade Flax NNX Gemma Sampling Inference doc
#4325 opened Oct 23, 2024 by 8bitmp3 Loading…
Fixing numbering in mnist_tutorial.md
#4311 opened Oct 18, 2024 by v1vekkumar Loading…
1 of 4 tasks
[nnx] flaxlib flatten
#4293 opened Oct 13, 2024 by cgarciae Loading…
Update Flax NNX vs JAX Transformations guide
#4286 opened Oct 10, 2024 by 8bitmp3 Loading…
Update Flax NNX Randomness
#4279 opened Oct 10, 2024 by 8bitmp3 Loading…
Stackless yashful
#4246 opened Oct 3, 2024 by copybara-service bot Loading…
proxy more methods for nnx.Variable
#4234 opened Sep 29, 2024 by njzjz Loading…
1 of 4 tasks
Utils for fp8 gated einsum
#4221 opened Sep 24, 2024 by wenscarl Loading…
Upgrade Flax NNX Filters doc
#4199 opened Sep 16, 2024 by 8bitmp3 Loading…
Upgrade Flax NNX Bridge API doc
#4198 opened Sep 16, 2024 by 8bitmp3 Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.