Implement JAX just-in-time compilation on Turboflow

turbo-sim / turboflow

Meanline model for performance prediction and preliminary design of turbomachinery

MIT License

11 stars 1 forks source link

Implement JAX just-in-time compilation on Turboflow #7

Open RoberAgro opened 1 month ago

RoberAgro commented 1 month ago

Implement just-in-time compilation in all the functions from the code Implementation should allow selecting whether or not to use JIT based on an user input