Implement Eq for TypeKind
Support more versions of `jit_function_apply`
Implement PartialEq for Field
Update tests