All modules for which code is available

• builtins
• functools
• jax._src.mesh
• jax._src.partition_spec
• jax._src.sharding_impls
• jaxlib.xla_extension
• jaxtyping._decorator
• ptyrodactyl.electrons.classes
• ptyrodactyl.electrons.forward
• ptyrodactyl.electrons.inverse
• ptyrodactyl.optics.lenses
• ptyrodactyl.tools.loss_functions
• ptyrodactyl.tools.optimizers
• ptyrodactyl.tools.parallel
• typing