tf2jax
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
109 stars
7 watching
11 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
Related projects:
Repository | Description | Stars |
---|---|---|
| Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
| A TensorFlow implementation of DeepLab using ResNet 101 architecture | 22 |
| A set of utilities for writing reliable JAX code | 797 |
| A tool for converting TensorFlow models to CoreML format for deployment on Apple devices. | 1,333 |
| An abstraction layer between various deep learning frameworks and your program. | 149 |
| A lightweight library for working with graph neural networks in jax. | 1,380 |
| A unified reshaping library for JAX and other frameworks. | 100 |
| A lightweight library for defining and training neural networks in TensorFlow. | 373 |
| This project focuses on manipulating 3D views using deep learning techniques. | 6 |
| An end-to-end deep learning project demonstrating various TensorFlow techniques and applications in image classification, GANs, text classification, and more. | 1,113 |
| An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
| A Python implementation of graph matching-based deep neural network fusion with applications to model ensemble and federated learning. | 18 |
| A Haskell framework for defining and compiling valid deep learning models to external frameworks like TensorFlow JS or Keras. | 101 |
| Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 252 |
| A library that enables distributed deep learning by partitioning tensors across processors in a mesh topology. | 1,597 |