Skip to content
Livermore Big Artificial Neural Network Toolkit
C++ Python Cuda CMake Shell
Branch: develop
Clone or download
Latest commit 4232883 Aug 23, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
bamboo
cmake Merge branch 'develop' of https://github.com/LLNL/lbann into fix-osx-… Aug 7, 2019
config Added the public key for the Travis CI builds Feb 21, 2017
containers Replacing deleted model in Singularity README. Feb 13, 2019
docs Updated the OS X build documents to reflect new bugfixes Aug 6, 2019
experiments Adding hacked bugfix for MPI_Init in MVAPICH to cycgan scripts. Apr 25, 2019
external Add tb image support (#1086) Jun 21, 2019
include Embedded Python session is destroyed during LBANN finalization. Aug 15, 2019
model_zoo use keyworded target_link_libraries Aug 20, 2019
python add the summarizer to the model class in the python front-end Aug 15, 2019
scripts Updated the spack environment to define the modules format (#1083) Jun 18, 2019
spack_environments Updated the spack environment to define the modules format (#1083) Jun 18, 2019
src silence gcc warning by initializing a variable Aug 19, 2019
superbuild add tmp and stamp dirs for openblas Aug 16, 2019
tests change the includes to use the most specific pb header possible Jul 31, 2019
tools Updated the copyright information. May 15, 2019
unit_test remove extraneous debugging output from CMake Feb 28, 2019
.astylerc First open source release of LBANN: Livermore Big Artificial Neural Jul 18, 2016
.gitignore added " *.mc Oct 19, 2017
.readthedocs.yml add readthedocs config Mar 22, 2019
.travis.yml Travis is failing to build doxygen files. Sep 9, 2017
CMakeLists.txt fix a cmake ordering issue Aug 7, 2019
CONTRIBUTORS First open source release of LBANN: Livermore Big Artificial Neural Jul 18, 2016
LICENSE Updated the copyright information. May 15, 2019
README.md Added a link to the publications at the top level markdown page. Mar 19, 2019
README_coding_style.txt First open source release of LBANN: Livermore Big Artificial Neural Jul 18, 2016
ReleaseNotes.txt Preparing release v0.99 May 15, 2019
lbann_license_template.txt Updated the copyright information. May 15, 2019

README.md

LBANN: Livermore Big Artificial Neural Network Toolkit

The Livermore Big Artificial Neural Network toolkit (LBANN) is an open-source, HPC-centric, deep learning training framework that is optimized to compose multiple levels of parallelism.

LBANN provides model-parallel acceleration through domain decomposition to optimize for strong scaling of network training. It also allows for composition of model-parallelism with both data parallelism and ensemble training methods for training large neural networks with massive amounts of data. LBANN is able to advantage of tightly-coupled accelerators, low-latency high-bandwidth networking, and high-bandwidth parallel file systems.

LBANN supports state-of-the-art training algorithms such as unsupervised, self-supervised, and adversarial (GAN) training methods in addition to traditional supervised learning. It also supports recurrent neural networks via back propagation through time (BPTT) training, transfer learning, and multi-model and ensemble training methods.

Building LBANN

The preferred method for LBANN users to install LBANN is to use Spack. After some system configuration, this should be as straightforward as

spack install lbann

More detailed instructions for building and installing LBANN are available at the main LBANN documentation.

Running LBANN

The basic template for running LBANN is

<mpi-launcher> <mpi-options> \
    lbann <lbann-options> \
    --model=model.prototext \
    --optimizer=opt.prototext \
    --reader=data_reader.prototext

When using GPGPU accelerators, users should be aware that LBANN is optimized for the case in which one assigns one GPU per MPI rank. This should be borne in mind when choosing the parameters for the MPI launcher.

More details about running LBANN are documented here.

Publications

A list of publications, presentations and posters are shown here.

Reporting issues

Issues, questions, and bugs can be raised on the Github issue tracker.

You can’t perform that action at this time.