AtomicGraphNets.jl implements graph-based models for machine learning on atomic systems, such as Crystal Graph Convolutional Neural Nets, in Julia. It makes use of the Flux ecosystem for model building and the JuliaGraphs ecosystem for graph representation and visualization, as well as adapting some features from GeometricFlux.
Documentation is in progress over here.
-
To install the latest tagged version, in your Julia REPL, do
]add AtomicGraphNets. However, you can also play with the latest version on themainbranch by skipping to step 2 and then doing]add /path/to/repowhere you replace the dummy path with the location of your clone. -
Clone this package to wherever you want to play.
-
Go and try out the example in examples/example1/ – it has its own README file with detailed instructions.
We welcome community contributions! Please refer to contribution guide for suggestions on how to go about things.
Many thanks to Dhairya Gandhi for helping out with some adjoints to actually make these layers trainable! :D