generics rework, stuff moved to fir for less shitty (arguably) handling. member func generics work too.
generics rework, stuff moved to fir for less shitty (arguably) handling. member func generics work too.

adrian17/flax,adrian17/flax,adrian17/flax