Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/mujocolab/mjlab
/ functions
Functions
2,812 in github.com/mujocolab/mjlab
⨍
Functions
2,812
◇
Types & classes
316
↳
Endpoints
15
Method
num_pairs
(self)
src/mjlab/entity/entity.py:497
Method
num_rays
Total number of rays (num_frames * num_rays_per_frame).
src/mjlab/sensor/raycast_sensor.py:590
Method
num_rays_per_frame
Number of rays per attachment frame.
src/mjlab/sensor/raycast_sensor.py:600
Method
num_sites
(self)
src/mjlab/entity/entity.py:477
Method
num_sites
(self)
tests/test_scene_entity_config.py:32
Method
num_targets
Number of targets. ``ctrl_ids`` is laid out as ``[pos..., vel...]``, each block of length ``num_targets``.
src/mjlab/actuator/builtin_actuator.py:148
Method
num_tendons
(self)
src/mjlab/entity/entity.py:481
Function
object_to_goal_distance
Distance vector from object to goal in base frame.
src/mjlab/tasks/manipulation/mdp/observations.py:39
Function
obs_A
(env)
tests/test_observation_history.py:557
Function
obs_B
(env)
tests/test_observation_history.py:561
Function
obs_func
(env)
tests/test_observation_noise.py:39
Function
obs_func
(env)
tests/test_observation_history.py:37
Function
obs_func
(env)
tests/test_observation_delay.py:37
Function
obs_func2
(env)
tests/test_observation_history.py:459
Function
obs_func2
(env)
tests/test_observation_delay.py:483
Function
obs_func_a
(env)
tests/test_observation_noise.py:282
Function
obs_func_b
(env)
tests/test_observation_noise.py:285
Function
obs_with_nan
(env)
tests/test_observation_nan_handling.py:35
Function
obs_with_neginf
(env)
tests/test_observation_nan_handling.py:410
Method
observation_space
(self)
src/mjlab/rl/vecenv_wrapper.py:36
Method
offset
Action offset.
src/mjlab/envs/mdp/actions/actions.py:130
Method
on_viewer_pause
Notify all command terms of viewer pause state change.
src/mjlab/managers/command_manager.py:191
Method
on_viewer_pause
(self, paused: bool)
src/mjlab/managers/command_manager.py:302
Method
on_viewer_pause
(self, paused: bool)
src/mjlab/tasks/tracking/mdp/commands.py:555
Function
open_stairs
( **overrides: Any, )
src/mjlab/terrains/config.py:183
Method
output_channels
Always ``None`` (output is flattened keypoints).
src/mjlab/rl/spatial_softmax.py:110
Method
output_dim
Total flattened output dimension (C * 2).
src/mjlab/rl/spatial_softmax.py:105
Function
pair_env
(device)
tests/test_domain_randomization.py:1810
Function
pair_friction
Randomize geom-pair friction overrides. Pair friction has 5 components: ``[tangent1, tangent2, spin, roll1, roll2]``. Default axis is 0 (tangent1
src/mjlab/envs/mdp/dr/pair.py:20
Method
pair_names
(self)
src/mjlab/entity/entity.py:455
Function
pd_gains
Randomize PD stiffness and damping gains. Args: env: The environment. env_ids: Environment IDs to randomize. If None, randomizes all. k
src/mjlab/envs/mdp/dr/actuator.py:30
Function
per_env_obs_func
(env)
tests/test_observation_delay.py:331
Method
per_world_default_fields
Fields with per-world defaults from mesh variant compilation.
src/mjlab/sim/sim.py:391
Function
perlin_noise
( **overrides: Any, )
src/mjlab/terrains/config.py:137
Method
physics_dt
Physics simulation step size.
src/mjlab/envs/manager_based_rl_env.py:261
Function
pole_angle_cos_sin
Cosine and sine of the pole hinge angle. Shape: [num_envs, 2].
src/mjlab/tasks/cartpole/cartpole_env_cfg.py:83
Method
prefixed_name
The sensor name with entity prefix if applicable.
src/mjlab/sensor/builtin_sensor.py:207
Method
prev_action
Raw policy output from the previous step, before per-term scale/offset. Shape: ``(num_envs, total_action_dim)``.
src/mjlab/managers/action_manager.py:120
Method
prev_prev_action
Raw policy output from two steps ago, before per-term scale/offset. Shape: ``(num_envs, total_action_dim)``.
src/mjlab/managers/action_manager.py:126
Method
primary_names
Primary names in the order they appear along the per-contact dim. Per-contact fields ([B, N, ...]) are laid out primary-major, so primary `pr
src/mjlab/sensor/contact_sensor.py:247
Method
process_actions
Process raw actions by applying scale, offset, and optional clip.
src/mjlab/envs/mdp/actions/actions.py:154
Method
process_actions
(self, actions: torch.Tensor)
src/mjlab/envs/mdp/actions/differential_ik.py:174
Function
process_docstring
Strip auto-generated dataclass docstrings (e.g. 'ClassName(*, ...)').
docs/conf.py:188
Function
process_signature
Suppress the ugly __init__ signature for dataclass Cfg classes.
docs/conf.py:180
Function
project_points
r"""Projects 3D points into 2D image plane. This project 3D points into a 2D image plane. The transformation is defined by the intrinsic matr
src/mjlab/utils/lab_api/math.py:1218
Function
projected_gravity
( env: ManagerBasedRlEnv, asset_cfg: SceneEntityCfg = _DEFAULT_ASSET_CFG, )
src/mjlab/envs/mdp/observations.py:38
Method
projected_gravity_b
Gravity vector projected into body frame. Shape (num_envs, 3).
src/mjlab/entity/data.py:584
Function
pseudo_inertia
r"""Physics-consistent inertial randomization via the pseudo-inertia matrix. Jointly randomizes ``body_mass``, ``body_ipos``, ``body_inertia``, and
src/mjlab/envs/mdp/dr/body.py:423
Function
push_by_setting_velocity
Push an entity by overwriting its root velocity with a sampled offset. This is an *instantaneous, mass-independent* kick: it adds a uniformly sampl
src/mjlab/envs/mdp/events.py:316
Method
qfrc_actuator
Forces produced by all actuators, mapped into joint space. For motors this is the commanded torque times the gear ratio. For position and vel
src/mjlab/entity/data.py:415
Method
qfrc_external
Forces on joints due to Cartesian wrenches applied to bodies. When a force or torque is applied to a body via ``xfrc_applied``, this property
src/mjlab/entity/data.py:426
Function
quat_apply_yaw
Rotate a vector only around the yaw-direction. Args: quat: The orientation in (w, x, y, z). Shape is (N, 4). vec: The vector in (
src/mjlab/utils/lab_api/math.py:673
Function
quat_box_plus
The box-plus operator (quaternion update) to apply an increment to a quaternion. Args: q: The initial quaternion in (w, x, y, z). Shape i
src/mjlab/utils/lab_api/math.py:608
Function
quat_env
(device)
tests/test_domain_randomization.py:500
Method
random
Generate a random RGBA color with specified alpha.
src/mjlab/utils/color.py:39
Function
random_orientation
Returns sampled rotation in 3D as quaternion. Args: num: The number of rotations to sample. device: Device to create tensor on.
src/mjlab/utils/lab_api/math.py:1299
Function
random_spread_boxes
( **overrides: Any, )
src/mjlab/terrains/config.py:167
Function
random_stairs
( **overrides: Any, )
src/mjlab/terrains/config.py:198
Function
random_yaw_orientation
Returns sampled rotation around z-axis. Args: num: The number of rotations to sample. device: Device to create tensor on. Re
src/mjlab/utils/lab_api/math.py:1319
Function
randomize_terrain
Randomize the sub-terrain for each environment on reset. This picks a random terrain type (column) and difficulty level (row) for each environmen
src/mjlab/envs/mdp/events.py:58
Method
raw_action
(self)
src/mjlab/managers/action_manager.py:67
Method
raw_action
Raw actions (before scale/offset).
src/mjlab/envs/mdp/actions/actions.py:135
Method
raw_action
(self)
src/mjlab/envs/mdp/actions/differential_ik.py:171
Method
record_post_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:39
Method
record_post_step
(self)
tests/test_recorder_manager.py:42
Method
record_post_step
(self)
tests/test_recorder_manager.py:86
Method
record_post_step
(self)
tests/test_recorder_manager.py:188
Method
record_post_step
(self)
tests/test_recorder_manager.py:256
Method
record_post_step
(self)
tests/test_recorder_manager.py:475
Method
record_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:36
Method
record_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:83
Method
record_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:185
Method
record_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:471
Method
remove
(self)
tests/test_variants.py:1296
Method
render
(self)
src/mjlab/viewer/offscreen_renderer.py:168
Method
render
(self)
src/mjlab/envs/manager_based_rl_env.py:484
Method
render_context
(self)
src/mjlab/sensor/sensor_context.py:97
Method
render_mode
(self)
src/mjlab/rl/vecenv_wrapper.py:32
Method
renderer
(self)
src/mjlab/viewer/offscreen_renderer.py:63
Function
repeat_array_kernel
( src: wp.array(dtype=Any), # type: ignore nelems_per_world: int, dst: wp.array(dtype=Any), # type: ig
src/mjlab/sim/randomization.py:10
Function
requires_model_fields
Mark an event function as requiring specific model fields expanded per-world. Fields listed here are registered in ``EventManager.domain_randomizat
src/mjlab/managers/event_manager.py:69
Method
reset
Reset the environment.
src/mjlab/utils/wrappers/video_recorder.py:81
Method
reset
Reset noise model state. Override in subclasses if needed.
src/mjlab/utils/noise/noise_model.py:26
Method
reset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/managers/curriculum_manager.py:95
Method
reset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/curriculum_manager.py:151
Method
reset
Resets the manager term.
src/mjlab/managers/manager_base.py:84
Method
reset
Resets the manager and returns logging info for the current step.
src/mjlab/managers/manager_base.py:119
Method
reset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/event_manager.py:216
Method
reset
(self, env_ids: torch.Tensor | None)
src/mjlab/managers/command_manager.py:241
Method
reset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/command_manager.py:320
Method
reset
Reset history buffers for specified environments. Args: env_ids: Environment indices to reset. If None, reset all environments.
src/mjlab/actuator/learned_actuator.py:122
Method
reset
Reset actuator state for specified environments. Resets delay buffers if present. Subclasses that override this should call ``super().reset(e
src/mjlab/actuator/actuator.py:365
Method
reset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/envs/mdp/events.py:584
Method
reset
Reset raw actions to zero for specified environments.
src/mjlab/envs/mdp/actions/actions.py:165
Method
reset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/envs/mdp/actions/differential_ik.py:257
Method
reset
Reset sensor state for specified environments. Invalidates the data cache. Override in subclasses that maintain internal state, but call `sup
src/mjlab/sensor/sensor.py:129
Method
reset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/sensor/contact_sensor.py:352
Method
reset
(self, env_ids: torch.Tensor | None = None, env=None)
tests/test_metrics_manager.py:25
Method
reset_environment
Extend BaseViewer.reset_environment to clear reward histories.
src/mjlab/viewer/native/viewer.py:445
Function
reset_joints_by_offset
( env: ManagerBasedRlEnv, env_ids: torch.Tensor | None, position_range: tuple[float, float], velocity_
src/mjlab/envs/mdp/events.py:279
← previous
next →
1,701–1,800 of 2,812, ranked by callers