Skip to content

Issues: patrick-kidger/equinox

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

174 Open 341 Closed
174 Open 341 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

Model trains but only if I dont JIT the step function?
#971 opened Mar 12, 2025 by Artur-Galstyan
Converter only applied when __post_init__ is defined for subclasses
#969 opened Mar 9, 2025 by gorold
WeightNorm causes unexpected PyTrees inequality feature New feature
#965 opened Mar 1, 2025 by yuanz271
8
Questions on eqx.filter_vmap: How to batch multiple function question User queries
#963 opened Feb 27, 2025 by timnotavailable
7
Slow Initialization question User queries
#961 opened Feb 26, 2025 by nstarman
9
What is the difference between jax.lax.stop_gradient and eqx.static_field=True ? question User queries
#958 opened Feb 24, 2025 by timnotavailable
7
BatchNorm initialization
#955 opened Feb 22, 2025 by ZagButNoZig
3
Truncated backprop using filtered-transformations question User queries
#954 opened Feb 21, 2025 by SNMS95
3
filter_vmap in_axes tree structure question User queries
#952 opened Feb 19, 2025 by vadmbertr
2
Should RoPE be included in embeddings?
#942 opened Feb 4, 2025 by lockwo
2
error_if and sharding
#939 opened Jan 27, 2025 by PhilipVinc
1
Adding another example documentation Improvements or additions to documentation
#936 opened Jan 23, 2025 by TugdualKerjan
1
Question: States, vmap-construction-of-layers and lax.scan'ing.
#934 opened Jan 16, 2025 by homerjed
7
filter_pmap traces function twice
#932 opened Jan 9, 2025 by garymm
4
Port of PureJaxRL to equinox
#926 opened Jan 6, 2025 by dorjeduck
5
Bounded while loop
#923 opened Jan 4, 2025 by lockwo
1
2
Improving eqx.nn.Embedding's performance
#920 opened Dec 24, 2024 by neel04
6
Question/recommendation about versioning question User queries
#918 opened Dec 23, 2024 by f0uriest
3
Replacement for equinox.internal.if_mapped? question User queries
#916 opened Dec 19, 2024 by mjo22
2
trouble deserializing a equinox tree leaf that has size [1,1]
#914 opened Dec 18, 2024 by csafta
1
using tfp.substrates.jax.optimizer.lbfgs_minimize with Equinox question User queries
#906 opened Dec 5, 2024 by raj-brown
6
jitting a module with segment sums question User queries
#903 opened Nov 28, 2024 by errhernandez
8
Speed of function that was jitted with eqx.filter_jit is slower than Flax's jit equivalent on CPU question User queries
#897 opened Nov 17, 2024 by Artur-Galstyan
1
Right way to created Masked CNN question User queries
#895 opened Nov 14, 2024 by ashutosh-dwivedi-e3502
1
Upsampling/Unpooling for Transpose Convolutions feature New feature
#894 opened Nov 10, 2024 by JadM133
10
ProTip! Find all open issues with in progress development work with linked:pr.