4 ms·

Check out Lux.jl and Reactant.jl for a modern attempt to create a Julia library with Jax’s XLA compiler backend!

• 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 ago
    Couldn’t agree more.