jaxline
AI summary
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
- stars
- 153
- forks
- 21
- watching
- 13
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
Meta learning experiment
A repository containing code for a meta-learning experiment on image datasets
JAX helper library
A set of utilities for writing reliable JAX code
Graph library
A lightweight library for working with graph neural networks in jax.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Deep learning framework bridge
An abstraction layer between various deep learning frameworks and your program.
DL experiment framework
A Python module for Deep Learning experiments on Audio and Text data combining classification, recommendation, and matrix factorization techniques.
google/paxml461
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
Android environment
A platform for defining reinforcement learning tasks on top of Android devices
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
Reshaper
A unified reshaping library for JAX and other frameworks.
Experiment framework
A framework for implementing agent-based experiments with cognitive processing and multi-agent interactions
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
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.
RL framework
A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jaxline.
