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
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
 chenxi116/tf-deeplab | A TensorFlow implementation of DeepLab using ResNet 101 architecture | 22 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 797 |
 tf-coreml/tf-coreml | A tool for converting TensorFlow models to CoreML format for deployment on Apple devices. | 1,333 |
 google-research/visu3d | An abstraction layer between various deep learning frameworks and your program. | 149 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
| google-deepmind/einshape | A unified reshaping library for JAX and other frameworks. | 100 |
| google-research/tf-slim | A lightweight library for defining and training neural networks in TensorFlow. | 373 |
 gamrix/cs231n_proj | This project focuses on manipulating 3D views using deep learning techniques. | 6 |
 burness/tensorflow-101 | An end-to-end deep learning project demonstrating various TensorFlow techniques and applications in image classification, GANs, text classification, and more. | 1,113 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
 thinklab-sjtu/gamf | A Python implementation of graph matching-based deep neural network fusion with applications to model ensemble and federated learning. | 18 |
 leopiney/tensor-safe | A Haskell framework for defining and compiling valid deep learning models to external frameworks like TensorFlow JS or Keras. | 101 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 252 |
 tensorflow/mesh | A library that enables distributed deep learning by partitioning tensors across processors in a mesh topology. | 1,597 |