API Reference

The public API is grouped by package area. Most users will start with sb3_extra_buffers.compressed.

sb3_extra_buffers	Extra buffer classes for Stable-Baselines3 with compression.
sb3_extra_buffers.compressed	Compressed rollout and replay buffers for Stable-Baselines3.
sb3_extra_buffers.recording	Game-state recording buffers for debugging and analysis.
sb3_extra_buffers.training_utils	Helpers for environment setup, evaluation, and buffer warm-up.
sb3_extra_buffers.vec_buf	Experimental vectorized rollout buffer wrappers.
sb3_extra_buffers.gpu_buffers	GPU-backed raw storage, compression, and SB3 buffers (experimental).

• Compressed Buffers
  • CompressedRolloutBuffer
  • CompressedDictRolloutBuffer
  • CompressedReplayBuffer
  • CompressedDictReplayBuffer
  • CompressedArray
  • DummyCls
  • find_smallest_dtype()
  • has_igzip()
  • has_numba()
  • init_jit()
  • find_buffer_dtypes()
  • Core classes
  • Helpers
  • Implementation modules
• Recording Buffers
  • BaseRecordBuffer
  • RecordBuffer
  • FramelessRecordBuffer
  • DummyRecordBuffer
  • Classes
  • Implementation modules
• Training Utilities
  • Atari environments
  • Evaluation
  • Buffer warm-up
• Vectorized Buffer Helpers
  • DummyVecRolloutBuffer
  • Implementation modules
• GPU Buffers
  • Overview
  • Metadata
  • Raw storage
  • Observation stores
  • Base helpers
  • Replay buffer
  • Rollout buffer
  • Utilities
  • Compression methods
  • Usage
  • Heap layout and compaction