distrax
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
536 stars
18 watching
32 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 788 |
| google-deepmind/functa | A repository containing code for a meta-learning experiment on image datasets | 149 |
| google-deepmind/optax | A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components. | 1,697 |
| google-deepmind/android_env | A platform for defining reinforcement learning tasks on top of Android devices | 1,019 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
| google-deepmind/einshape | A unified reshaping library for JAX and other frameworks. | 99 |
| google-deepmind/tree | A library for working with nested data structures in Python | 945 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 389 |
 jppbsi/libdeep | An open-source C library providing tools and components for developing artificial intelligence-based techniques using deep learning. | 23 |
| google-deepmind/rlax | A Python library providing reinforcement learning building blocks for implementing agent policies and functions | 1,263 |
| google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 105 |
| google-deepmind/recurrentgemma | An implementation of a fast and efficient language model architecture | 607 |
 google/objax | An open-source machine learning framework designed to simplify and accelerate research | 771 |
 sheffieldml/deepgp | Software implementing probabilistic models for uncertainty estimation in data analysis | 90 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |