Java EE is pretty much dead in the traditional it was marketed and spoken of. Trendy things in the Java ecosystem nowadays are frameworks such as Spring Boot and MicroProfile (which took the good stuff from the old Java EE. Think JAX-RS for REST API's etc).
Django works with Python and in the AI space you'll want to iterate as quickly as possible, so I think that's probably going to be the preferred choice for beginners.