Applied Scientist
CurrentMeenthttps://github.com/kc-ml2/meent- Developed 'Meent': code for optical simulations and optimization. -- https://github.com/kc-ml2/meent -- implemented on Numpy, JAX and PyTorch. -- Supports Automatic Differentiation (AD) which returns gradient of loss per each input variable. -- with the help of JAX pmap, Meent support parallelization(CPU and GPU).