Skip to content

Commit ed63093

Browse files
authored
Merge pull request #299 from FluxML/doc
Add roadmap
2 parents 3dd2820 + 425863a commit ed63093

File tree

1 file changed

+23
-6
lines changed

1 file changed

+23
-6
lines changed

README.md

Lines changed: 23 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,12 @@ Suggestions, issues and pull requsts are welcome.
2121

2222
## Features
2323

24-
* Extend Flux deep learning framework in Julia and compatible with Flux layers.
25-
* Support of CUDA GPU with CUDA.jl
26-
* Integrate with existing JuliaGraphs ecosystem
27-
* Support generic graph neural network architectures
28-
* Variable graph inputs are supported. You use it when diverse graph structures are prepared as inputs to the same model.
29-
* Integrate GNN benchmark datasets (WIP)
24+
* Extending Flux deep learning framework in Julia and seamlessly integration with regular Flux layers.
25+
* Support of CUDA GPU with CUDA.jl and mini-batched training leveraging advantages of GPU
26+
* Integration with existing JuliaGraphs ecosystem
27+
* Support Message-passing and graph network architectures
28+
* Support of static graph and variable graph strategy. Variable graph strategy is useful when training the model over diverse graph structures.
29+
* Integration of GNN benchmark datasets
3030

3131
### Featured Graphs
3232

@@ -69,3 +69,20 @@ evalcb() = @show(accuracy(train_X, train_y))
6969

7070
Flux.train!(loss, ps, train_data, opt, cb=throttle(evalcb, 10))
7171
```
72+
73+
## Roadmap
74+
75+
To achieve geometric deep learning raised by Bronstein *et al*, 5G fields of deep learning models will be supported in GeometricFlux.jl. For details, you could check the [geometric deep learning official website](https://geometricdeeplearning.com/).
76+
77+
5G including the following fields:
78+
79+
* **Graphs** and Sets
80+
* including classical GNN models and networks over sets.
81+
* Transformer models are regard as a kind of GNN with complete graph, and you can check [chengchingwen/Transformers.jl](https://github.com/chengchingwen/Transformers.jl) for more details.
82+
* **Grids** and Euclidean spaces
83+
* including classical convolutional neural networks, multi-layer perceptrons etc.
84+
* for operators over functional spaces of regular grid, you can check [SciML/NeuralOperators.jl](https://github.com/SciML/NeuralOperators.jl) for more details.
85+
* **Groups** and Homogeneous spaces
86+
* including a series of equivariant/invariant models.
87+
* **Geodesics** and Manifolds
88+
* **Gauges** and Bundles

0 commit comments

Comments
 (0)