3 ms·
Sometimes it really feels like JAX should be a more general DSL that can embed in other languages, as long as you have some way to turn structures into pytrees
by catgary 2y ago
Sometimes it really feels like JAX should be a more general DSL that can embed in other languages, as long as you have some way to turn structures into pytrees and functions into jaxpr’s.
- mccoyb 2y agoCouldn’t agree more.