MuJoCo Warp (MJWarp), dibangun di atas NVIDIA Warp, memungkinkan model MuJoCo yang kompatibel berjalan pada skala GPU, sehingga memungkinkan pengembangan ratusan atau ribuan dunia simulasi independen dalam satu panggilan. Arsitektur ini menggeser fokus dari meminimalkan latensi untuk satu lingkungan ke meningkatkan throughput agregat, yang menguntungkan pembelajaran penguatan dan pengambilan sampel skala besar.

  • MJWarp mengimplementasikan jalur fisika MuJoCo menggunakan NVIDIA Warp, mengompilasi kernel CUDA untuk memajukan keadaan simulasi pada GPU NVIDIA.
  • Kerangka kerja ini mendukung pekerjaan paralel eksplisit, array perangkat, dan peluncuran kernel yang dapat dikomposisikan, dengan mode diferensiasi opsional dan eksekusi deterministik.
  • Penyetelan kinerja melibatkan pengambilan grafik CUDA untuk langkah-langkah berulang dan penyesuaian ukuran buffer kontak dan kendala seperti nconmax dan njmax.
  • Migrasi dari MuJoCo CPU memerlukan pencocokan laju simulasi dan validasi hasil terhadap baseline, seperti lengan SO-101 yang menyusun kubus.

Pendekatan ini membantu pengguna menskalakan alur kerja simulasi robotika dengan memanfaatkan paralelisme GPU untuk pengumpulan pengalaman yang efisien dalam alur pelatihan.