Exception raised when parsing output from a command fails.

A DataProto is a data structure that aims to provide a standard protocol for data exchange between functions. It contains a batch (TensorDict

Agent's return value.

A DataProto is a data structure that aims to provide a standard protocol for data exchange between functions. It contains a batch (TensorDict

A class that keeps all conversation history.

Python wrapper for the simulator written in Scala. The methods that are being wrapped can be found in simulator/src/main/scala/scienceworld/runtim

Registry for storing and building classes.

An LLM for generating texts from given prompts and sampling parameters. This class includes a tokenizer, a language model (possibly distributed

Agent's criticism.

From Multi-Head Attention module https://github.com/jadore801120/attention-is-all-you-need-pytorch

Agent's action to take.

Sharding manager to support data resharding when using FSDP + Ulysses

Sharding manager to support data resharding when using FSDP + Ulysses

Used to count mfu during training loop Example: flops_counter = FlopsCounter(config) flops_achieved, flops_promised = flops_

Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`LlamaDecoderLayer`] Args: config: LlamaConfig

We assume the dataset contains a column that contains prompts and other information

A wrapper of textEnv for models. Returns valid actions at each step of the game.

Word-level segment information

The model can behave as an encoder (with only self-attention) as well as a decoder, in which case a layer of cross-attention is added between

Processes an answer similar to Eval AI copied from https://github.com/facebookresearch/mmf/blob/c46b3b3391275b4181567db80943473a8

We assume the dataset contains a column that contains prompts and other information

Note that this trainer runs on the driver process on a single CPU/GPU node.

Define a resource pool specification. Resource pool will be initialized first. Mapping

This is an in-memory SFTDataset

Aligned sentence segment

Allows to handle custom placeholders 'title_color' and 'message_no_color'. To use this formatter, make sure to pass 'color', 'title' as log e

Processes an answer similar to Eval AI copied from https://github.com/facebookresearch/mmf/blob/c46b3b3391275b4181567db80943473a8

A checkpoint manager that saves and loads - model - optimizer - lr_scheduler - extra_states in a SPMD way. We save

A checkpoint manager that saves and loads - model - optimizer - lr_scheduler - extra_states in a SPMD way. We save

Linear interpolation between initial_p and final_p over schedule_timesteps. After this many timesteps pass final_p is returned. :para

Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`PhiDecoderLayer`] Args: config: PhiConfig

Define a resource pool specification. Resource pool will be initialized first. Mapping

an extension of ai2thor.controller.Controller for ALFRED tasks

Audio Feature Extractor Class Main Functions: 1. Load and process single audio files 2. Extract word-level features using Triton

DataProtoFuture aims to eliminate actual data fetching on driver. By doing so, the driver doesn't have to wait for data so that asynchronous

DataProtoFuture aims to eliminate actual data fetching on driver. By doing so, the driver doesn't have to wait for data so that asynchronous

The implementation of a simple digit completion task. The prompt is a sequence of numbers with fixed difference. The task is to complete the

Used to count mfu during training loop Example: flops_counter = FlopsCounter(config) flops_achieved, flops_promised = flops_

A basic LSTM cell.

A checkpoint manager that saves and loads - model - optimizer - lr_scheduler - extra_states in a SPMD way. We save

A memory buffer is a contiguous torch tensor that may combine multiple tensors sharing with the underlying memory. It must have a unique type

A memory buffer is a contiguous torch tensor that may combine multiple tensors sharing with the underlying memory. It must have a unique type

A client class for interacting with Gemini 2.5 Pro model through OpenRouter API. This class handles the complete workflow of: 1. Acc