Skip to content

Add conditonal primitive support

Compare
Choose a tag to compare
@zombie-einstein zombie-einstein released this 26 Aug 19:34
· 19 commits to main since this release
a232c3d

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