it has to have a .zip extension

That should be fine, a Pex file can be whatever file extension. See https://github.com/pantsbuild/pants/blob/732d94424873e56a1eb7b540c382a4f6dde1aec3/src/python/pants/backend/awslambda/python/rules.py#L78-L81