Add conditonal primitive support
Add support for generating sub-graphs representing conditional JAX functions, and a general restructure of the codebase.
Add support for generating sub-graphs representing conditional JAX functions, and a general restructure of the codebase.