# TODO - [ ] Tests - [ ] Node - [ ] Context - [ ] Node structs - [ ] functions to create nodes - [ ] nodes should take in context and state - [ ] each node should have an id - [ ] run - [ ] train - [ ] init state