You can further condense the code through using pm.math.sigmoid rather than defining it manually
pm.math.sigmoid