issues
search
patrick-kidger
/
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache License 2.0
2.12k
stars
142
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
An array-like defined with `eqx.Module` does not support `jnp.asarray` with the `__array__` method
#898
gautierronan
closed
17 hours ago
4
Speed of function that was jitted with eqx.filter_jit is slower than Flax's jit equivalent on CPU
#897
Artur-Galstyan
opened
4 days ago
1
Optax.Multisteps problems, callable set to None
#896
louisfabrice13
closed
1 week ago
2
Right way to created Masked CNN
#895
ashutosh-dwivedi-e3502
opened
1 week ago
1
Upsampling/Unpooling for Transpose Convolutions
#894
JadM133
opened
1 week ago
1
Unexpected results with __subclasshook__
#893
SimonKoop
closed
1 week ago
2
Incompatibility with jax 0.4.34 (and 0.4.35)
#892
Prsodk
closed
2 weeks ago
3
pylint unable to detect ``__init__`` signature correctly when base class has ``__init__``
#891
ji8er
opened
2 weeks ago
1
Add Generic Type Hints to tree_at Function
#890
jsonmona
closed
2 weeks ago
5
Bug in how eqx.Module interacts with jax.tree.map
#889
xaviergonzalez
opened
2 weeks ago
2
A potential typo? Or should we do `eqx.filter(model, eqx.is_array)` here in the RNN example?
#887
DigitalPig
closed
3 weeks ago
4
Is Equinox compatible with Flax?
#886
jeertmans
closed
4 weeks ago
4
ValueError when updating a State value from "weak_f32" to "f32" with jax >= 0.4.33
#885
elinorbgr
opened
1 month ago
2
ConvTranspose1d bug
#884
TugdualKerjan
opened
1 month ago
1
Version bump v0.11.8
#883
patrick-kidger
closed
1 month ago
0
AbstractVar property: interaction bug with static type-checker
#882
EtaoinWu
opened
1 month ago
1
Fix introductory CNN example in recent JAX.
#881
patrick-kidger
closed
1 month ago
0
issue with the introductory example "CNN on MNIST"
#880
pasq-cat
closed
1 month ago
1
Replicating jax's export feature to cache filter_jitted functions
#879
aeftimia
opened
1 month ago
4
Allow Python versions higher than 3.9
#878
norpadon
closed
1 month ago
3
Improves docs for `tree_at` and `tree_check`.
#877
patrick-kidger
closed
1 month ago
2
Fix computation in normalization layers
#876
AakashKumarNain
closed
1 month ago
5
custom trace estimation resulting in 2 * expected result
#875
nahid18
closed
1 month ago
1
chore(docs): notify about `tree_at` bypassing checks
#874
jeertmans
closed
1 month ago
4
Add memorax link to readme
#873
smorad
opened
1 month ago
1
`tree_at` silently skips `__check_init__`
#872
jeertmans
closed
1 month ago
2
More fixes for JAX 0.4.34
#871
patrick-kidger
closed
1 month ago
0
Entering `while_loop` when condition is `False`
#870
dkweiss31
closed
1 month ago
4
Issue with Equinox compatibility after `jax==0.4.34` update
#869
Qazalbash
closed
1 month ago
4
Forward compatibility with future versions of JAX errors.
#868
patrick-kidger
closed
1 month ago
1
JAX 0.4.33 changes how errors work, which broke the nice error_if messages. Hopefully this should be cross-version compatible.
#867
patrick-kidger
closed
1 month ago
0
How to mark parameters as trainable or not?
#866
tomsch420
opened
1 month ago
5
Callables not marked static prevents dynamic call
#865
NeilGirdhar
closed
1 month ago
8
PyTorch Hooks in Equinox?
#864
samuelstevens
opened
1 month ago
2
"JAX array is set a static" warning is raised unwantedly
#863
gautierronan
opened
1 month ago
5
pyright can't tell that Module fields are frozen
#862
garymm
closed
1 month ago
4
`eval_shape` incompatible with deserializing directly to host due to `__assert_same`
#861
colehaus
opened
1 month ago
3
Added RoPE Offset + Test
#860
Artur-Galstyan
opened
2 months ago
0
InitVar and pickle break PyTreeDef equality
#859
HGangloff
opened
2 months ago
1
Problems with access to class variables in __init_subclass__ during instance creation for equinox Modules
#858
SimonKoop
closed
1 week ago
2
InitVar and deepcopy break PyTreeDef equality
#857
HGangloff
opened
2 months ago
2
Hopefully compatibility with JAX 0.4.32
#856
patrick-kidger
closed
2 months ago
0
Access to all hidden states in recurrent neural networks
#855
qtomcatq
closed
2 months ago
2
Issue with jax and jaxlib versions 0.4.32 and 0.4.33
#854
dkweiss31
closed
1 month ago
4
optax.MultiSteps
#853
haydn-jones
closed
2 months ago
6
Add overloads to eqx.combine
#852
NeilGirdhar
closed
2 months ago
1
specify mkdocs-autorefs version
#851
garymm
closed
2 months ago
1
v0.11.6
#850
patrick-kidger
closed
2 months ago
0
Now filtering pure_callback's messages via logs rather than at stderr.
#849
patrick-kidger
closed
2 months ago
0
Add [test] and update CONTRIBUTING
#848
dlwh
closed
2 months ago
4
Next