> The disadvantage is that it's [...] not trying to be a fully-fledged competitive neural network framework Perhaps this is an unpopular opinion, but to me thi

> The disadvantage is that it's [...] not trying to be a fully-fledged competitive neural network framework Perhaps this is an unpopular opinion, but to me this is an advantage. JAX is a library upon which you can build a neural-network framework -- or a framework for something else. It seems clear that we're still figuring out the Right Way to write code that defines a neural network. The fact that JAX lets you write different, competing libraries/APIs/DSLs -- however you want to think of it -- lets us innovate more freely.