Currently we support all the behaviors that require data dependent shapes (currently just the unique_* functions, nonzero, and boolean indexing).
It would be useful to have some sort of mode to disable these so people could test against libraries that don't have them.