flax

56f69104 - Automaticlaly remove non-init attributes from dataclass hash/eq/repr