Using JAX to accelerate our research

Open source Using JAX to accelerate our research Published 4 December 2020 Authors David Budden, Matteo Hessel DeepMind engineers accelerate our research by building tools, scaling up algorithms, and creating challenging virtual and physical worlds for training and testing artificial intelligence (AI) systems. As part of this work, we constantly evaluate new machine learning libraries and framewor

[misshiki]

“NumPy APIに加えて、JAXには、次のような機械学習研究のサポートに役立つ、構成可能な関数変換の拡張可能なシステムが含まれています。”微分、ベクトル化、JITコンパイル。DeepMindでの活用が広がっており要注目。