QDax: A Library for Quality-Diversity and Population-based Algorithms with Hardware Acceleration