equinox

equinox copied to clipboard

Hey Patrick, I was wondering if there was a way to set the default behavior for the filter functions in some global manner? I am currently building an [fully differentiable...

LouisDesdoigts

added use_ceil to pooling

1

Corresponding issue: https://github.com/patrick-kidger/equinox/issues/175 - Added `use_ceil` parameter to `Pool` - Added a check to make sure garbage values are not pooled if `padding > kernel/2`.

paganpasta

Corresponds to discussion : https://github.com/patrick-kidger/equinox/issues/136 - Adds `_ordered_tree_map` - Some doc fixes to highlight cases of deserialisation.

paganpasta

Adding ceil_mode to nn.Pool?

1

Hi, Is it possible to support torch's `ceil_mode` like functionality in Pooling? There is a possible* one sided padding for each dimension as a result of using `ceil` instead of...

paganpasta

`eqx.nn.Conv2d` produces slightly different results than `torch.nn.Conv2d`

4

Hi, I wanted to check if it is possible for the two operations to produce identical results. I found that `jnp.isclose` assertion fails on the output when comparing `torch` vs...

paganpasta

Adds utils for calculating attention weights and multihead attention without projection layers.

1

Fixes #142

anshradh

Fixes #166.

patrick-kidger

Feature Request: Add default parameter values to docstring

5

Currently the docstrings for `Module` subclasses don't display the default argument values when using `?`. Not sure if there is an automated way to display this, or to update all...

jenkspt

Recommended way to partially load weights into a model ?

32

Hi, Is there a recommended way to partially load weights to a new model ? At the moment I have an inept solution which uses the fact that `fc` attribute...

paganpasta

Guideline for using `static_fields`?

24

Hi, Most of the `nn.Modules` ([MLP](https://github.com/patrick-kidger/equinox/blob/a89d5b486d13588caffc095f172a2ec39fd68278/equinox/nn/composed.py#L26)) use `static_fields` for, well seemingly static attributes. In the documentation it is stated that `static_field` should be used rarely. Is there a do's and...

paganpasta