jaxline
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
152 stars
13 watching
21 forks
Language: Python
last commit: 12 months ago
Linked from 1 awesome list
amrzv/awesome-colab-notebooksRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 105 |
| google-deepmind/functa | A repository containing code for a meta-learning experiment on image datasets | 149 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 788 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 google-research/visu3d | An abstraction layer between various deep learning frameworks and your program. | 148 |
 sergiooramas/tartarus | A Python module for Deep Learning experiments on Audio and Text data combining classification, recommendation, and matrix factorization techniques. | 101 |
 google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 457 |
| google-deepmind/android_env | A platform for defining reinforcement learning tasks on top of Android devices | 1,019 |
| google/jaxopt | An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning. | 933 |
| google-deepmind/einshape | A unified reshaping library for JAX and other frameworks. | 99 |
 dwarfmaster/babel2 | A framework for implementing agent-based experiments with cognitive processing and multi-agent interactions | 6 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 389 |
| 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 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |