Hey Wojciech , if possible please consider adding typescript/javascript support to Flax.