distrax
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
538 stars
18 watching
32 forks
Language: Python
last commit: almost 2 years agoLinked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 797 |
| google-deepmind/functa | A repository containing code for a meta-learning experiment on image datasets | 150 |
| 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,730 |
| google-deepmind/android_env | A platform for defining reinforcement learning tasks on top of Android devices | 1,029 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
| google-deepmind/einshape | A unified reshaping library for JAX and other frameworks. | 100 |
| google-deepmind/tree | A library for working with nested data structures in Python | 948 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
 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,272 |
| google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 109 |
| google-deepmind/recurrentgemma | An implementation of a fast and efficient language model architecture | 613 |
 google/objax | An open-source machine learning framework designed to simplify and accelerate research | 772 |
 sheffieldml/deepgp | Software implementing probabilistic models for uncertainty estimation in data analysis | 91 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |