tf2jax
AI summary
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
- stars
- 109
- forks
- 11
- watching
- 7
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
DeepLab model
A TensorFlow implementation of DeepLab using ResNet 101 architecture
JAX helper library
A set of utilities for writing reliable JAX code
Model converter
A tool for converting TensorFlow models to CoreML format for deployment on Apple devices.
Deep learning framework bridge
An abstraction layer between various deep learning frameworks and your program.
Graph library
A lightweight library for working with graph neural networks in jax.
Reshaper
A unified reshaping library for JAX and other frameworks.
Neural network library
A lightweight library for defining and training neural networks in TensorFlow.
View transformer
This project focuses on manipulating 3D views using deep learning techniques.
TensorFlow tutorial
An end-to-end deep learning project demonstrating various TensorFlow techniques and applications in image classification, GANs, text classification, and more.
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
Deep Neural Network Fusion Library
A Python implementation of graph matching-based deep neural network fusion with applications to model ensemble and federated learning.
Model compiler
A Haskell framework for defining and compiling valid deep learning models to external frameworks like TensorFlow JS or Keras.
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
tensorflow/mesh1.6K
Tensor processing library
A library that enables distributed deep learning by partitioning tensors across processors in a mesh topology.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions tf2jax.
