camber.llm

Library for workloads in large-language models.

import camber.llm

Methods

create_job

Creates a CamberJob using the LLM engine.

Args

command: str

Base command in string form

e.g. “python3 train.py” where train.py is the script to be executed

engine_size: Optional[str]

Size of engine, one of XMICRO, MICRO, XXSMALL, XSMALL, SMALL, MEDIUM, LARGE.

Default is XSMALL.

with_gpu: Optional[bool]

Whether to use GPU, defaults to False

tags: Optional[Union[str, List]]

Tags to help identify the job, can be string or list of strings

Returns

CamberJob

The CamberJob object representing the created job.

delete_job

Deletes an LLM based CamberJob from your account. Supply the id of the CamberJob.

Args

job_id: Union[int, str]

The id of the CamberJob object to delete

e.g. 1234

get_job

Retrieves an LLM based CamberJob from CamberCloud. Supply the id of the CamberJob.

Args

job_id: Union[int, str]

The id of the CamberJob object to retrieve

e.g. 1234

Returns

CamberJob

The retrieved CamberJob object.

list_jobs

Lists all LLM based CamberJobs in CamberCloud. Optionally supply the tags to filter the jobs.

Args

tags: Optional[Union[str, List]]

Tags to help identify the job.

e.g. ["tag1", "tag2"]

Returns

List[CamberJob]

The list of CamberJob objects.