AWS Batch has added support for GPU scheduling to G3, G3s, and G4 instance families, in addition to existing support for GPU scheduling on P2 and P3 families. 

from Recent Announcements https://aws.amazon.com/about-aws/whats-new/2019/11/aws-batch-supports-gpu-scheduling-g3-g3s-g4-instance-families/