“High performance machine learning with JAX“ - Mat Kelcey (PyConline AU 2021)

(Mat Kelcey) JAX provides an elegant interface to XLA with automatic differentiation allowing extremely high performance machine learning on modern accelerators; all from within Python. In this talk we’ll give an overview to fundamentals of JAX and an intro to some of the libraries being developed on top. python, pycon, australia, programming, conference, technical, pyconline, developers, panel, sessions, libraries, frameworks, community, sysadmins, students, education, data, science Videos licensed as CC-BY-NC-SA 4.0 PyCon AU is the national conference for the Python programming community, bringing together professional, student and enthusiast developers, sysadmins and operations folk, students, educators, scientists, statisticians, and many others besides, all with a love for working with Python. PyCon AU informs the country’s Python developers with presentations, tutorials and panel sessions by experts and core developers of Python, as well as the lib