KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #typescript #developer-tools #claude-code #ai-agent #codex #deepseek-harness #ai-agents #cli #react #mcp

hoomd-tf

★ 33 Open GitHub ↗

A plugin that allows the use of Tensorflow in Hoomd-Blue for GPU-accelerated ML+MD

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# HOOMD-TF

[![status](https://joss.theoj.org/papers/5d1323eadec82aabe86c65a403ff8f90/status.svg)](https://joss.theoj.org/papers/5d1323eadec82aabe86c65a403ff8f90)
[![Documentation Status](https://readthedocs.org/projects/hoomd-tf/badge/?version=latest)](https://hoomd-tf.readthedocs.io/en/latest/?badge=latest)
[![Build Stats](https://github.com/ur-whitelab/hoomd-tf/workflows/tests/badge.svg)](https://github.com/ur-whitelab/hoomd-tf/actions)

This plugin enables the use of TensorFlow in a [HOOMD-blue](http://glotzerlab.engin.umich.edu/hoomd-blue/) simulation to compute quantities like forces and collective variables and do learning while running a simulation. You may also use it without hoomd-blue to process trajectories via [MDAnalysis](https://www.mdanalysis.org/). Please see [here for documentation](https://hoomd-tf.readthedocs.io/en/latest) for install and usage instructions.

HOOMD-TF can be used for a variety of tasks such as online force-matching, online machine learning in HOOMD-blue simulations, and arbitrary collective variable calculations using TensorFlow tensor operations. Because both HOOMD-blue and TensorFlow are GPU-accelerated, HOOMD-TF was designed with speed in mind, and minimizes latency with a GPU-GPU communication scheme. Of particular interest, HOOMD-TF allows for online machine learning with early termination, rather than the more tradditional batch learning approach for MD+ML.

HOOMD-TF includes several utility functions as convenient built-ins, such as:
* RDF calculation
* EDS Biasing (See [this paper](https://www.tandfonline.com/doi/full/10.1080/08927022.2019.1608988))
* Coarse-Grained simulation force matching

In addition to all these, the TensorFlow interface of HOOMD-TF makes implementing arbitrary ML models as easy as it is in TensorFlow, by exposing the HOOMD-blue neighbor list and particle positions to TensorFlow. This enables GPU-accelerated tensor calculations, meaning arbitrary collective variables can be treated in the TensorFlow model framework, as long as they can be expressed as tensor operations on particle positions or neighbor lists.

# Tutorials

See [example notebooks here](https://nbviewer.jupyter.org/github/ur-whitelab/hoomd-tf/tree/master/examples/) to learn about what HOOMD-TF can do.

# Prerequisites

The following packages are required to compile:

tensorflow >= 2.3
    hoomd >= 2.6
    tbb-devel (only for hoomd if installed with conda)

tbb-devel is required when using the
HOOMD-blue conda release. It is not automatically installed when
installing HOOMD-blue, so use `conda install -c conda-forge tbb-devel`
to install. The TensorFlow version should be TensorFlow 2.3 release.
It is recommended you install via pip.

# Citation

Please use the following citation:

> HOOMD-TF: GPU-Accelerated, Online Machine Learning in the HOOMD-blue Molecular Dynamics Engine. R Barrett, M Chakraborty, DB Amirkulova,
> HA Gandhi, G Wellawatte, and AD White (2020) *Journal of Open Source Software* doi: 10.21105/joss.02367

hoomd-tf

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README