Add Flax dummy objects
# What does this PR do?

Following the first JAX models, this PR adds the dummy objects to make sure the library always has the same objects available.
cc @mfuntowicz for information

