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
Filter Primitive Transpose for trace of linear operators
#847
quattro
closed
2 months ago
4
Add detail to make_with_state doc.
#846
garymm
closed
2 months ago
3
fix link from gan to diffusion
#845
garymm
closed
2 months ago
1
Infinite recursion with the stderr wrapper thing
#844
dlwh
closed
1 month ago
8
Allow StateIndex to be passed dynamically
#843
NeilGirdhar
closed
1 month ago
13
StateIndex is a Module, but not a PyTree
#842
NeilGirdhar
closed
1 month ago
5
How to decide module's learnable parameters at initialisation
#841
itk22
closed
2 months ago
2
Accessing self during __post_init__ seems to result in a differently registered pytree
#840
garymm
closed
2 months ago
2
Question/Discussion: How should you accomplish state/memory in Equinox?
#839
NeilGirdhar
closed
2 months ago
6
Equinox Decorators transforming scalar functions into arrays
#838
cwoolfo1
closed
2 months ago
6
build: bump up precommit hooks
#837
knyazer
closed
2 months ago
3
fix: RoPE embeddings do not preserve dtypes
#836
knyazer
closed
2 months ago
8
eqx.error_if errors even when predicate is false if input array is empty
#835
allen-adastra
opened
2 months ago
5
Fixes cooperative multiple inheritance with `__post_init__`
#834
patrick-kidger
closed
2 months ago
3
Fix __post_init__ detection in Module
#833
NeilGirdhar
closed
2 months ago
2
Equinox seems to break cooperative multiple inheritance
#832
NeilGirdhar
closed
2 months ago
4
Add exact spectral norm feature
#831
lockwo
opened
2 months ago
1
Expose attention weights' head dimension
#830
neel04
opened
2 months ago
5
Allow splitting to any number of partitions
#829
francois-rozet
closed
1 month ago
0
Add `_FilteredStderr.flush`.
#828
patrick-kidger
closed
2 months ago
0
v0.4.11.5 Breaks Pycharm Debugger Console
#827
adam-hartshorne
closed
2 months ago
11
[Feat] Add fused QKV projection
#826
neel04
closed
2 months ago
6
Native Model Parallellism in equinox
#825
neel04
opened
2 months ago
9
Spliting a module in more than 2 partitions
#824
francois-rozet
closed
1 month ago
7
Remove dead ruff option
#823
NeilGirdhar
closed
2 months ago
1
Fix MyPy's view of _ModuleMeta
#822
NeilGirdhar
closed
2 months ago
5
Add editorconfig
#821
NeilGirdhar
closed
2 months ago
7
MyPy doesn't understand Module
#820
NeilGirdhar
closed
2 months ago
2
Make _DinstinctTuple inherit from tuple
#819
danielward27
closed
2 months ago
1
_DistinctTuple is not a tuple causing errors in tree_at
#818
danielward27
closed
2 months ago
1
debug.assert_max_traces is not helpful when the difference is due to weak_type
#817
garymm
closed
2 months ago
1
Minimal Transformer encoder using MultiheadAttention and RoPE
#816
davidaknowles
closed
2 months ago
3
PositionalSharding tensor of different ranks: use sharding.replicate!
#815
davidaknowles
closed
2 months ago
3
Type errors with Module
#814
NeilGirdhar
closed
2 months ago
2
Question about efficiency of higher-order derivatives when defining custom jvp
#813
leonard-gleyzer
opened
2 months ago
1
Comparable PixelShuffle functionality?
#812
mathDR
closed
2 months ago
2
Non-default `__init__` with `default_factory` throws error.
#811
samuelstevens
closed
2 months ago
3
Spectral Norm Features
#810
lockwo
opened
3 months ago
1
Adding a kernel constraint
#809
the-wind-is-rising
closed
3 months ago
2
Add Tensor class
#808
mr-raccoon-97
closed
3 months ago
3
[Question/Maybe Issue] Is there a version of Jax.linearize in equinox
#807
krm9c
closed
3 months ago
3
[Question] Modifying the Static Variable in a Model
#806
ahahn2813
opened
3 months ago
5
Remove polyfill
#805
ColCarroll
closed
3 months ago
3
Update parallelism.ipynb
#804
garymm
closed
3 months ago
1
Runtime errors inside eqx.filter_jit are now very readable.
#803
patrick-kidger
closed
3 months ago
0
Version bump
#802
patrick-kidger
closed
3 months ago
0
eqx.filter_hessian gives incorrect result for jax.nn.relu
#801
SimonKoop
closed
1 week ago
1
Disable static arrays
#800
lockwo
closed
3 months ago
17
Added the offset to RoPE embedding and fixed the pre-commit pyright
#799
Artur-Galstyan
closed
2 months ago
6
Realistic scenario in which jax array is static
#798
lockwo
closed
3 months ago
4
Previous
Next