Not sure if this is exactly what you had in mind, but if you're up for experimenting it's worth a try:
https://cooking.nytimes.com/recipes/12363-lemon-pudding-cake
I've made it with almond milk and it was really good. And a non-dairy butter to grease the pan, of course. The texture is sort of like a molten chocolate lava cake, but not as heavy.