Skip to content

[api]

Model and Vector abstractions submodule.

This submodules exports the building blocks of the Model and Vector APIs:

  • Model: Abstract class defining an API to interface a ML model.
  • Vector: Abstract class defining an API to manipulate (sets of) data arrays.
  • VectorSpec: Metadata container to specify a Vector for its (un)flattening.
  • register_vector_type: Decorate a Vector subclass to make it buildable with Vector.build.