einshape
AI summary
Reshaper
A unified reshaping library for JAX and other frameworks.
- stars
- 100
- forks
- 6
- watching
- 7
- 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
Graph library
A lightweight library for working with graph neural networks in jax.
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
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.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Language model
An implementation of a fast and efficient language model architecture
Data navigator
A library for working with nested data structures in Python
Android environment
A platform for defining reinforcement learning tasks on top of Android devices
Social Learning Test Suite
Assesses generalization of multi-agent reinforcement learning algorithms to novel social situations
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Prob dist lib
A library of probability distributions and bijectors with a focus on readability, extensibility, and compatibility with existing frameworks.
RL builder
A Python library providing reinforcement learning building blocks for implementing agent policies and functions
Deep learning framework bridge
An abstraction layer between various deep learning frameworks and your program.
Efficient LLM
A large language model with improved efficiency and performance compared to similar models
Featured in 1 awesome list
Each link jumps to the spot where the list mentions einshape.
