Take a look at the code examples, looks like Python & Numpy + Haskell had a baby. And yes, it is developed by the creators of PyTorch (the irony) and some of the guys from Tensorflow, Jax and S4TF at Google Research & DeepMind. Read their paper here, it's pretty interesting. [1] It may never become an actual production language but it is many times more interesting than both S4TF and Julia.

[1] https://openreview.net/pdf?id=rJxd7vsWPS