Awesome JAX-based Models

Discover a variety of ready-made models and projects built with JAX that support machine learning research and data science. Learn about implementations ranging from Fourier feature networks to reinforcement learning, quantum chemistry, and protein modeling, providing practical resources to deepen your JAX expertise.