GitLab

* Support  periodic table Indexing in builtin models

* flake8

* more

* fix

* fix cuda

* Add element names to ANIModel

* nc trainer

Add helper module to convert species from element id in periodic table to 0, 1, 2, 3, ... format (#396)

* Init

* fix test

* flake8

* try fix

* Fix stupidity of len(self)

* Allow passing pbc and cell to sequential

* revert

* Simplify ase interface

* fix

* fix

* fix

* flake8

* Update ase.py

* fix

* Use namedtuple to improve API

* improve

* Use modern type annotations for nn.py

* flake8

* Don't hard code ensemble size

* flake8

* fix

* fix device type

* Update utils.py

* Update utils.py

fix typo!

* add test for scripted ensemble

* use torchani.nn.Sequential

* change OrderedDict to module list

* fix

* fix nn.py

* try more fix

* try

* more

* more

* fix more

* rename

* bring ensemble back

* make ANIModel iterable

* make aev,model compatible with jit

* add type annotation to nn

* flake8 fix

* refactor AEVComputer

* fix doc

* an example with padding

* use Optional type instead of padding

* fix

* fix

* make pbc and cell keyword arguments in test_aev

* fix

* make pbc and cell keyword arguments in ase

* fix

* fix

* fix dtype

* fix

* aev_computer dtype to double

* change test files to have aev_computer with keyword argument

* fix JIT types

* add TestAEVJIT

* fix LGTM alerts

* fix TestAEVJIT

* Update aev.py

workaround for dtype in `torch.arange`

* More arange bugs

* Even more arange

* fix LGTM alert