我正试图在JAX编码的环境中训练DQN代理，但训练器的初始化在第一次尝试重置环境时失败(JAX类型错误无效(。在进入调试过程之前，我曾想过寻找示例项目，但我什么都找不到，所以我想知道是否不可能与它们对接。
目前RLlib中没有现成的支持jax-envs的示例。我认为主要功能应该在那里，只是从来没有用Jax-envs进行过测试，以找出功能需求和缺失的部分。我鼓励你尝试一下，并在github上的ray问题跟踪器上为它创建一个功能请求，如果你有带宽，请尝试一下并做出贡献。您还可以让RL团队了解使RLlib能够在JAX环境中工作所缺少的功能。
javascript python java c# php android html jquery c++ css ios sql mysql arrays asp.net json python-3.x ruby-on-rails .net sql-server django objective-c excel regex ruby linux ajax iphone xml vba spring asp.net-mvc database wordpress string postgresql wpf windows xcode bash git oracle list vb.net multithreading eclipse algorithm macos powershell visual-studio image forms numpy scala function api selenium