araffin / sbx

SBX: Stable Baselines Jax (SB3 + Jax)

araffin/sbx Issues

[Feature Request] Dict Obs Spaces Support
Updated 17 days ago2
self.key is never updated
Closed 2 months ago2
[Bug] TQC Entropy Coefficient
Closed 3 months ago2
[Bug] TQC Hyperparameter optimization: Results do not match the reference. This is likely a bug/unexpected loss of precision.
Updated 3 months ago2
[Feature Request] Passing custom activation functon in policy_kwargs
Closed 3 months ago2
[Feature Request] Recurrent policies
Updated 3 months ago2
Mujoco XLA - MJX Integration
Closed 3 months ago1
[Question] MaskablePPO support
Closed 3 months ago1
[Question] Why is fps much lower than CPU if using GPU
Closed 3 months ago1
[Question] TypeError when exporting a model to PyTorch in SBX
Closed 3 months ago3
[Bug] example supplied in readme crashing
Closed 3 months ago4
[Question] Speedup compared to SB3
Closed 3 months ago1
[Bug] AttributeError: module 'tensorflow.python.util.tf_inspect' has no attribute 'Parameter'
Closed 3 months ago9
[Question] `MultiInputPolicy` not supported (DroQ)
Updated 4 months ago1
[Feature Request] Multi-Discrete action spaces for PPO
Closed 4 months ago3
[Enhancement] Support for large gradient_steps in SAC
Closed 5 months ago2
SBX becomes super slow when number of cpus are limited
Closed 5 months ago2
[Feature Request] Support Optax Optimizer Schedules
Closed 5 months ago4
[Feature Request] Update type annotations
Closed 5 months ago
crash when using a custom network architecture
Closed 5 months ago4
Custom env with FrameStack wrapper causes invalid actions to be passed to `env.step`
Closed 8 months ago2
[Question] Extending sbx algorithms (e.g via a callback)
Closed 10 months ago2
[Question] is it possible cython use this sbx?
Closed a year ago4
Adding Feature Extractors
Updated 2 years ago1