I've been thinking a lot about trying to use functional dialect of Python like Coconut[0] and Hy[1] w/ JAX so I can write functional DL code. Glad to see functorch[3] as PyTorch is the library I have the most experience with. [0] http://coconut-lang.org/ [1] https://docs.hylang.org/en/alpha/ [2] https://github.com/google/jax [3] https://github.com/pytorch/functorch
See also, Hissp: https://github.com/gilch/hissp