Cognitive Tasks

A modular, composable framework for constructing cognitive tasks for neural-network training and computational neuroscience simulations.

What you’ll find here

• A quickstart showing how to sample single trials and JIT/vmap-friendly batches from a pre-built task.

• A building-blocks tutorial covering features, declarative phases, encoders, label helpers, and both the instance-based and class-based Task definition patterns — plus vector-output tasks, conditional control flow, and custom encoders.

• A variable-length trial sequences tutorial that lays out what works today (If / Switch / While, TruncExp / UniformDuration), the current vmap-fixed-length limitation, the planned padding-plus-mask API, and workarounds you can use right now.

For the full list of pre-built tasks, encoders, phases, and utilities, see the API reference.

• Tutorial 1: Quickstart with braintools.cogtask
• Tutorial 2: Building custom tasks
• Tutorial 3: Variable-length trial sequences