JAX MD: A Framework for Differentiable Atomistic Physics

preview_player
Показать описание
Sam Schoenholz, Google Brain
  1. Physics Meets ML
Рекомендации по теме
JAX MD: A

JAX MD: A Framework for Differentiable Atomistic Physics

JAX MD: A

JAX MD: A Framework for Differentiable Atomistic Physics

What is JAX?

What is JAX?

What is the

What is the Jax Deep Learning Framework?

Neural Networks in

Neural Networks in Equinox (JAX DL framework) with Optax

Google's Jax MD

Google's Jax MD (Molecular Dynamics)

Google's AutoGrad +

Google's AutoGrad + Tensorflow's XLA Linear Algebra Compiler = JAX

Numpy on the

Numpy on the GPU? Speeding up Simple Machine Learning Algorithms with JAX

Introduction to JAX

Introduction to JAX

JAX Talk: Diffrax

JAX Talk: Diffrax

JAX Course -

JAX Course - 4. JAX Implementation of DALL·E

Intro to JAX

Intro to JAX

Experimenting with JAX

Experimenting with JAX | AI Professor Improvises Chess Programming #16

JAX Crash Course

JAX Crash Course - Accelerating Machine Learning code!

'High performance machine

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

Image Classification with

Image Classification with JAX and FLAX | PyImageSearch Bonus Lesson

Magical NumPy with

Magical NumPy with JAX - 01 Loopless Loops

Automatic differentiation in

Automatic differentiation in scientific programming with jax

JAX: Accelerated Machine

JAX: Accelerated Machine Learning Research via Composable Function Transformations in Python

MicroProfile: Optimizing Java

MicroProfile: Optimizing Java EE for a microservices architecture

Magical NumPy with

Magical NumPy with JAX - 00 Introduction

Strange JRAPH -

Strange JRAPH - Deep Mind's GNN Library for Graph Neural Networks (w/ JAX)

Intro to Jax/Flax

Intro to Jax/Flax (Taha Bouhsine)

JAX Course -

JAX Course - 3. Future of ML research in JAX/Flax