Open atiyo opened 2 months ago
The use of checkify.check can alter the return type of jitted functions, so it has been proposed to investigate chex (https://github.com/google-deepmind/chex) to replace the checkify.check calls in https://github.com/pasqal-io/horqrux/compare/feature/psr_on_all_gates
checkify.check
Also, it might be more streamlined in general to replace custom pytrees, e.g. in Primitive and Parametric among others with chex.dataclasses.
Primitive
Parametric
chex.dataclass
The use of
checkify.checkcan alter the return type of jitted functions, so it has been proposed to investigate chex (https://github.com/google-deepmind/chex) to replace thecheckify.checkcalls in https://github.com/pasqal-io/horqrux/compare/feature/psr_on_all_gatesAlso, it might be more streamlined in general to replace custom pytrees, e.g. in
PrimitiveandParametricamong others withchex.dataclasses.