Robert Boschgmbh

Shaojie Bai

Devin T Willmott

Jeremy Zico Kolter

Jonathan Mailoa

Mordechai C Kornbluth

Vladlen Koltun

Jeremy Zieg Kolter

João D Semedo

Filipe Condessa

Yash Savani

A computer-implemented method for a machine learning (ML) system includes receiving a first image frame and a second frame from a sensor, wherein the first and second image frames are time series data, determining a first flow state and a first latent state of the first image frame, determining a Deep Equilibrium Model (DEQ) based fix point solution via a root finding method based on the first flow state, the first latent state, and a layer function to obtain an estimated flow and latent state, receiving a third image frame, wherein the second and third image frames are time series data, determining the fix point solution via the root finding method based on the estimated flow state, the estimated latent state, and layer function to obtain an updated flow state and updated latent state, and outputting the updated flow state.

Deep equilibrium flow estimation

A computer-implemented method for a classification and training a neural network includes receiving input at the neural network, wherein the input includes a plurality of resolution inputs of varying resolutions, outputting a plurality of feature tensors for each corresponding resolution of the plurality of resolution inputs, fusing the plurality of feature tensors utilizing upsampling or down sampling for the vary resolutions, utilizing an equilibrium solver to identify one or more prediction vectors from the plurality of feature tensors, and outputting a loss in response to the one or more prediction vectors.

System and method for multiscale deep equilibrium models

A simulation includes converting a molecular dynamics snapshot of elements within a multi-element system into a graph with atoms as nodes of the graph; defining a matrix such that each column of the matrix represents a node in the graph; defining a distance matrix according to a set of relative positions of each of the atoms; iterating through the GTFF using an attention mechanism, operating on the matrix and augmented by incorporating the distance matrix, to pass hidden state from a current layer of the GTFF to a next layer of the GTFF; performing a combination over the columns of the matrix to produce a scalar molecular energy; making a backward pass through the GTFF, iteratively calculating derivatives at each of the layers of the GTFF to compute a prediction of force acting on each atom; and returning the prediction of the force acting on each atom.

Graph transformer neural network force field for prediction of atomic forces and energies in molecular dynamic simulations

Growing community of inventors

Pittsburgh, PA, United States of America

Shaojie Bai