graph_framework: A graph computation framework that supports auto differentiation. It is designed to allow domain scientists to create cross platform GPU accelerated code and embed those codes in existing legacy tools. It is designed for the domain of physics problems where there same physics is applied to a large ensemble of independent systems.
This framework enables:
- Portability to Nvidia, AMD, and Apple GPUs and CPUs.
- Abstraction of the physics from the compute.
- Auto Differentiation.
- Embedding in C, C++, and Fortran codes.
The compute kernels created have strong scaling to multiple devices
and the best throughput on both GPUs and CPUs compared to other frameworks like MLX and JAX
graph_framework-docs Documentation for the graph_framework.
To get started clone this repository using the command.
git clone https://github.com/ORNL-Fusion/graph_framework.git
For instructions to build the code consult the build system documentation. This framework uses a cmake based build system and requires the NetCDF-C library.

