{"library":"distrax","type":"library","category":null,"description":"Distrax is a DeepMind library offering a comprehensive collection of probability distributions and bijectors, tightly integrated with JAX for high-performance numerical computation, automatic differentiation, and GPU acceleration. It provides a flexible API for constructing complex probabilistic models and is widely used within the JAX ecosystem for research and development. The library typically follows JAX's release cadence for compatibility, with frequent updates for new features and bug fixes. Current version is 0.1.7.","language":"python","status":"active","version":"0.1.7","tags":["JAX","probability","distributions","machine-learning","deepmind","probabilistic-programming"],"last_verified":"Tue May 26","install":[{"cmd":"pip install distrax","imports":["from distrax import Categorical","from distrax import Normal","from distrax import Distribution","from distrax import Transformed","from distrax import Bijector"]}],"homepage":null,"github":"https://github.com/google-deepmind/distrax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/distrax/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":50,"avg_install_s":16,"avg_import_s":4.4,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/distrax/compatibility"}}