Mahmoud Asem

Class fields and instance fields behavior

3

Hello, Patrick currently, data classes allow fields to be either class or instance variables for example ```python @dc.dataclass class Test: a:int = 1 def __init__(self) -> None: pass Test().__dict__ #...

# Bug Report I was reading through your excellent documentation, and I found what I think is bug (on jax side IMO) allowing non-hashable `KeyArray` as a `static_field`.  [Relevant...

kernex operate on views of the first function array argument, it is useful to add argnum/argname to select which argument to operate on in a similar fashion in jax transformation

- Dilation can be added by changing the view indices generation step value.

`at` can match paths using regex pattern, use this feature and write a sample example on how set sharding in a similar sense how the new keras API implements it...

Draft PR https://github.com/google/jax/pull/18137 points to a plan to deprecate `jax.scipy.ndimage` including `map_coordinates`, In which`serket.image` subpackage and couple of other internal functions in revision (e.g. deformable convolution) rely on.

[docs] checklist

1

- [ ] Toy model conversion from `PyTorch` - [ ] Toy model conversion from `keras` (i.e. show lazy approach) - [x] orbax for checkpointing