Elixir (https://elixir-lang.org) and its Nx stack (https://github.com/elixir-nx), maybe ?