distrax
AI summary
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
- stars
- 538
- forks
- 32
- watching
- 18
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
JAX helper library
A set of utilities for writing reliable JAX code
Meta learning experiment
A repository containing code for a meta-learning experiment on image datasets
Gradient optimizer library
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.
Android environment
A platform for defining reinforcement learning tasks on top of Android devices
Graph library
A lightweight library for working with graph neural networks in jax.
Reshaper
A unified reshaping library for JAX and other frameworks.
Data navigator
A library for working with nested data structures in Python
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Deep learning toolkit
An open-source C library providing tools and components for developing artificial intelligence-based techniques using deep learning.
RL builder
A Python library providing reinforcement learning building blocks for implementing agent policies and functions
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
Language model
An implementation of a fast and efficient language model architecture
google/objax772
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
Uncertainty model
Software implementing probabilistic models for uncertainty estimation in data analysis
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Featured in 1 awesome list
Each link jumps to the spot where the list mentions distrax.
