Skip to content

Pull requests: google/flax

New pull request New
95 Open 2,249 Closed
95 Open 2,249 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add support for jax.sharding.PartitionSpec.UNCONSTRAINED in logical specification
#3902 opened May 4, 2024 by copybara-service bot Loading…
[linen] test jit caching with state updates
#3900 opened May 3, 2024 by cgarciae Loading…
Add a get_inputs_kernel_bias method to the DenseGeneral layer.
#3896 opened May 1, 2024 by copybara-service bot Loading…
1
[nnx] fix mypy and pytype
#3894 opened Apr 30, 2024 by cgarciae Loading…
1
[linen] fix DenseGeneral init
#3893 opened Apr 30, 2024 by copybara-service bot Loading…
v0.8.4
#3891 opened Apr 30, 2024 by cgarciae Loading…
[nnx] fix iter_nodes
#3889 opened Apr 28, 2024 by cgarciae Loading…
1
Fix exception cause in configurations.py
#3888 opened Apr 26, 2024 by cool-RR Loading…
Let num_heads=1 by default in MultiHeadAttention.
#3883 opened Apr 24, 2024 by carlosgmartin Loading…
[nnx] add Module pytree_experimental static test
#3864 opened Apr 17, 2024 by cgarciae Loading…
Make ConvLSTMCell.__call__ infer the number of features from its input, like the other recurrent cells.
#3862 opened Apr 16, 2024 by carlosgmartin Loading…
1
Minor typo fix in flax_fundamentals/flax_basics.ipynb
#3852 opened Apr 12, 2024 by albanie Loading…
1
Minor docstring fix for flax.linen.Module
#3851 opened Apr 12, 2024 by albanie Loading…
Minor fix to docstring for flax.core.variables.py
#3850 opened Apr 12, 2024 by albanie Loading…
Minor docstring fix for flax.linen.module.variable
#3849 opened Apr 12, 2024 by albanie Loading…
Minor doc formatting fixes for flax.linen.Module.init_with_output
#3848 opened Apr 12, 2024 by albanie Loading…
1 task done
Added three tab option to sphinx directive codediff and added testing for first tab
#3847 opened Apr 11, 2024 by chiamp Loading…
@chiamp
3
Flax NNX message in ToC
pull ready
#3845 opened Apr 11, 2024 by 8bitmp3 Loading…
@cgarciae
Fix minor typos in docstring and markdown file
#3844 opened Apr 11, 2024 by dantp-ai Loading…
1 of 4 tasks
1
remove tree_map deprecation warning
#3836 opened Apr 9, 2024 by chiamp • Draft
1
@chiamp
[linen] enable separate initializers for out layer in MultiHeadDotProductAttention
pull ready
#3835 opened Apr 9, 2024 by cgarciae Loading…
1
Add **init_kwargs to self.param type annotation.
#3832 opened Apr 8, 2024 by copybara-service bot Loading…
1
Bump pillow from 10.2.0 to 10.3.0 in /examples/vae
dependencies Pull requests that update a dependency file
#3821 opened Apr 3, 2024 by dependabot bot Loading…
1
[linen] allows multiple compact methods
#3808 opened Apr 1, 2024 by cgarciae Loading…
1
ensure doctest runs on NNX
#3797 opened Mar 29, 2024 by chiamp Loading…
@chiamp
2
ProTip! Add no:assignee to see everything that’s not assigned.