replace the big `validate_config` logic in the `computation_graph` class with something that uses [jsonschema](https://pypi.org/project/jsonschema/) or [pydantic](https://docs.pydantic.dev/latest/)