patrick-kidger/equinox: Callable PyTrees and filtered transforms => neural networks in JAX. https://docs.kidger.site/equinox/

799
STARS
16
WATCHERS
45
FORKS
21
ISSUES

equinox's Language Statistics

patrick-kidger's Other Repos

Star history of equinox
Issue history of equinox

equinox Recent Issues

Issue Title State Comments Created Date Updated Date Closed Date
Batchnorm example doesn't work open 3 2022-09-22 2022-09-27 -
Counting the number of parameters in an equinox module. open 1 2022-09-19 2022-09-27 -
[Contribution] Port equinox version of NeRF model open 2 2022-09-13 2022-09-27 -
Easier syntax for `eqx.tree_at()` closed 6 2022-09-08 2022-09-27 2022-09-15
eqx.Module interaction with optax closed 2 2022-09-07 2022-09-27 2022-09-07
Performance of `filter_pmap` open 4 2022-09-07 2022-09-27 -
`eqx.filter_grad` Question closed 1 2022-09-04 2022-09-27 2022-09-04
too strict version req? closed 1 2022-09-04 2022-09-16 2022-09-05
Return value from serialisation closed 4 2022-09-02 2022-09-27 2022-09-06
How to get outputs from an intermediate layer? closed 10 2022-08-31 2022-09-27 2022-09-06
Create "advanced tricks" section of the docs open 0 2022-08-31 2022-09-27 -
Using grad with BatchNorm fails at the second iteration closed 2 2022-08-30 2022-09-27 2022-08-30
Couple of suggestions for `nn.Sequential`. open 5 2022-08-26 2022-09-16 -
filter_pmap + Dropout/Pooling breaks under distributed training closed 2 2022-08-21 2022-09-16 2022-08-24
Why uniform bias initializer? open 9 2022-08-19 2022-09-26 -
Globally set default filter functions? closed 4 2022-08-12 2022-09-16 2022-08-13
Adding ceil_mode to nn.Pool? closed 2 2022-08-09 2022-09-16 2022-08-31
`eqx.nn.Conv2d` produces slightly different results than `torch.nn.Conv2d` open 4 2022-08-08 2022-09-16 -
Error upon importing equinox closed 1 2022-08-03 2022-09-16 2022-08-03
Default serialisation fails for `BatchNorm`. closed 10 2022-08-03 2022-09-16 2022-08-31
Missing #output in the example of StateIndex. closed 1 2022-07-31 2022-09-16 2022-08-01
Syntax for using `BatchNorm` with `filter_jit` and `filter_grad_value`? closed 3 2022-07-30 2022-09-16 2022-07-30
Adding eqx.nn.Lambda ? closed 1 2022-07-28 2022-09-16 2022-07-30
Guideline for using `static_fields`? open 14 2022-07-26 2022-09-16 -
Make `nn.Sequential` indexable closed 1 2022-07-26 2022-09-16 2022-07-28
Recommended way to count number of compilations for a `filter_jit` function? closed 2 2022-07-26 2022-09-16 2022-07-26
Borrowing code for doc generation. closed 1 2022-07-24 2022-09-30 2022-07-25
Updating `operation` in `equinox.nn.Pool` to `static_field`? closed 4 2022-07-24 2022-09-25 2022-07-25
Feature Request: Add default parameter values to docstring open 5 2022-07-23 2022-09-16 -
Adding eqx.nn.Sequential for chaining layers? closed 1 2022-07-23 2022-09-16 2022-07-23
How to parallelize model training? closed 2 2022-07-23 2022-09-16 2022-07-23
VGG support or eqxvision repository? closed 2 2022-07-22 2022-09-16 2022-08-29
Feature Request: add attention utility functions like flax closed 3 2022-07-21 2022-09-16 2022-08-31
support for common networks VGG, ResNet, Xception closed 1 2022-07-20 2022-09-16 2022-07-20
Best Dataloader practices open 6 2022-07-16 2022-09-16 -
Recommended way to partially load weights into a model ? open 28 2022-07-15 2022-09-16 -
static fields don't work with filter_grad(arg=...)? closed 6 2022-07-07 2022-09-16 2022-07-09
`channels` may not be unspecified in `GroupNorm`, despite what the docs say closed 1 2022-07-07 2022-09-16 2022-07-24
Automatically compile dependant Equinox modules closed 6 2022-06-30 2022-09-16 2022-07-01
Freeze only some entries in a parameter array? closed 2 2022-06-29 2022-09-16 2022-06-30
Support for adaptive average pooling? closed 8 2022-06-23 2022-09-16 2022-07-23
Out-of-place updates with custom `__init__()` constructor open 3 2022-06-22 2022-09-16 -
"TypeError: unhashable type: 'JVPTracer'" workaround? closed 4 2022-06-19 2022-09-16 2022-06-23
MultiheadAttention fails with jit closed 2 2022-06-16 2022-09-16 2022-06-17
Lack of `out=` breaks use of `filter_pmap(fn)` when output of `fn` involves non-JAX types closed 6 2022-06-15 2022-09-16 2022-07-05
Thoughts on adding equinox.vision.models similar to torchvision.models ? closed 2 2022-06-15 2022-09-16 2022-06-15
Equinox equivalent of FuncTorch's make_functional open 4 2022-06-13 2022-09-16 -
Unbound axis name in filter_pmap + lax.pmean closed 1 2022-06-13 2022-09-25 2022-06-14
Feature request: Equinox version of `optax.incremental_update(...)` closed 2 2022-06-13 2022-09-16 2022-06-13
Standard MLP (equinox.nn.MLP) does not work with `apply_updates` function closed 3 2022-06-13 2022-09-16 2022-06-13