Francesco Zanetta

Optional `fastmath` compilation?

12

Big thanks to the authors for this amazing library. As an xarray user I am expecting to benefit from it quite a lot :) While running some micro benchmarks on...

Cannot assign arrays to dataclass fields in `nnx`

3

When instantiating a `nnx.dataclass` module, if the input to a `param_field` (or any `variable_field` actually) is a jax Array, a ValueError is raised because the value is assigned to the...

When I create a new parameter I can pass extra keyword arguments as metadata: ```python import dataclasses from flax.experimental import nnx foo = nnx.Param( 1.0, domain = "positive_real", ) print(foo)...

## Description Introduces `flax` / `nnx` as backend for the library. ## Tasks - [x] pass all standard tests - [x] pass integration tests - [x] check and eventually update...

Nonstationary RFF

3

## Type of changes - [ ] Bug fix - [x] New feature - [ ] Documentation / docstrings - [ ] Tests - [ ] Other ## Checklist -...

Inconsistent level ordering when loading data from FDB source

1

### What happened? When loading data from the FDB source, depending on whether the `lazy=True` flag is used or not, the ordering of the fieldlists (and derived xarray objects) based...

Understanding the behavior of `group_by`

1

Hello, I am using the `fdb` source for accessing reanalysis data here at MeteoSwiss. In my processing pipeline, I use `group_by("param")` on the fields coming from my request. This is...

Support for ICON grid

2

### Goal If my information is correct, development to support ICON unstructured grid is already in the pipeline in earthkit-(geo,data), coming also from [eckit](https://github.com/search?q=repo%3Aecmwf%2Feckit+icon&type=code). I am opening this issue so...

Custom codec information is silently dropped when writing kerchunk references

6

I encountered this bug when working on a custom GRIB parser in https://github.com/MeteoSwiss/icon-ch-vzarr. When trying to serialize virtual datasets as kerchunk references, the custom codec information is silently dropped. The...