Physics ∩ ML Seminars |
Events During the Week of September 5th through September 12th, 2021
Monday, September 6th, 2021
- No events scheduled
Tuesday, September 7th, 2021
- No events scheduled
Wednesday, September 8th, 2021
- JAX MD: A Framework for Differentiable Atomistic Physics
- Time: 11:00 am - 12:15 pm
- Place: Online Seminar: Please sign up for our mailing list at www.physicsmeetsml.org for zoom link
- Speaker: Sam Schoenholz, Google Brain
- Abstract: I will talk about JAX MD, a software package for performing differentiable physics simulations with a focus on molecular dynamics. JAX MD includes a number of physics simulation environments, as well as interaction potentials and neural networks that can be integrated into these environments without writing any additional code. Since the simulations themselves are differentiable functions, entire trajectories can be differentiated to perform meta-optimization. These features are built on primitive operations, such as spatial partitioning, that allow simulations to scale to hundreds-of-thousands of particles on a single GPU. My talk will include an introduction to the JAX software package www.github.com/google/jax. If you are interested in trying out JAX MD, it is available at github.com/google/jax-md.
- Host: Gary Shiu
Thursday, September 9th, 2021
- No events scheduled
Friday, September 10th, 2021
- No events scheduled