{"data":{"node":{"slug":"jax-ml-jax","name":"jax","tagline":"Composable transformations of Python+NumPy programs","github_url":"https://github.com/jax-ml/jax","owner":"jax-ml","repo":"jax","owner_avatar_url":"https://avatars.githubusercontent.com/u/58486408?v=4","primary_language":"Python","stars":36085,"forks":3714,"topics":["jax"],"archived":false,"github_pushed_at":"2026-08-02T08:08:23+00:00","maintenance_label":"Very active","url":"https://www.graphcanon.com/tools/jax-ml-jax","markdown_url":"https://www.graphcanon.com/tools/jax-ml-jax.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/jax-ml-jax","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=jax-ml-jax"},"categories":[{"slug":"inference-serving","name":"Inference & Serving","url":"https://www.graphcanon.com/categories/inference-serving","markdown_url":"https://www.graphcanon.com/categories/inference-serving.md","api_url":"https://www.graphcanon.com/api/graphcanon/categories/inference-serving"},{"slug":"model-training","name":"Model Training","url":"https://www.graphcanon.com/categories/model-training","markdown_url":"https://www.graphcanon.com/categories/model-training.md","api_url":"https://www.graphcanon.com/api/graphcanon/categories/model-training"}],"tags":[{"slug":"compilation","name":"compilation"},{"slug":"differentiation","name":"differentiation"},{"slug":"gpu","name":"gpu"},{"slug":"python","name":"python"},{"slug":"tpu","name":"tpu"}],"edges":[{"type":"depends_on","direction":"in","explanation":"Keras supports JAX as a backend for executing machine learning models with JAX's high-performance computation infrastructure.","successor_context":null,"tool":{"slug":"keras-team-keras","name":"keras","tagline":"Deep Learning for humans","github_url":"https://github.com/keras-team/keras","owner":"keras-team","repo":"keras","owner_avatar_url":"https://avatars.githubusercontent.com/u/34455048?v=4","primary_language":"Python","stars":64217,"forks":19746,"topics":["data-science","deep-learning","jax","machine-learning","neural-networks","python","pytorch","tensorflow"],"archived":false,"github_pushed_at":"2026-08-01T11:15:08+00:00","maintenance_label":"Very active","url":"https://www.graphcanon.com/tools/keras-team-keras","markdown_url":"https://www.graphcanon.com/tools/keras-team-keras.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/keras-team-keras","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=keras-team-keras"}}],"neighbours":[{"slug":"tensorflow-tensorflow","name":"tensorflow","tagline":"An Open Source Machine Learning Framework for Everyone","github_url":"https://github.com/tensorflow/tensorflow","owner":"tensorflow","repo":"tensorflow","owner_avatar_url":"https://avatars.githubusercontent.com/u/15658638?v=4","primary_language":"C++","stars":196758,"forks":75773,"topics":["deep-learning","deep-neural-networks","distributed","machine-learning","ml","neural-network","python","tensorflow"],"archived":false,"github_pushed_at":"2026-08-03T06:01:06+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/tensorflow-tensorflow","markdown_url":"https://www.graphcanon.com/tools/tensorflow-tensorflow.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/tensorflow-tensorflow","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=tensorflow-tensorflow","shared_categories":["model-training"]},{"slug":"pytorch-pytorch","name":"pytorch","tagline":"Tensors and Dynamic neural networks in Python with strong GPU acceleration","github_url":"https://github.com/pytorch/pytorch","owner":"pytorch","repo":"pytorch","owner_avatar_url":"https://avatars.githubusercontent.com/u/21003710?v=4","primary_language":"Python","stars":102144,"forks":28650,"topics":["autograd","deep-learning","gpu","machine-learning","neural-network","numpy","python","tensor"],"archived":false,"github_pushed_at":"2026-08-03T06:00:50+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/pytorch-pytorch","markdown_url":"https://www.graphcanon.com/tools/pytorch-pytorch.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/pytorch-pytorch","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=pytorch-pytorch","shared_categories":["model-training","inference-serving"]},{"slug":"lightning-ai-pytorch-lightning","name":"pytorch-lightning","tagline":"Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.","github_url":"https://github.com/Lightning-AI/pytorch-lightning","owner":"Lightning-AI","repo":"pytorch-lightning","owner_avatar_url":"https://avatars.githubusercontent.com/u/58386951?v=4","primary_language":"Python","stars":31267,"forks":3768,"topics":["ai","artificial-intelligence","data-science","deep-learning","machine-learning","python","pytorch"],"archived":false,"github_pushed_at":"2026-08-03T01:13:09+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/lightning-ai-pytorch-lightning","markdown_url":"https://www.graphcanon.com/tools/lightning-ai-pytorch-lightning.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/lightning-ai-pytorch-lightning","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=lightning-ai-pytorch-lightning","shared_categories":["model-training","inference-serving"]},{"slug":"optuna-optuna","name":"optuna","tagline":"A hyperparameter optimization framework","github_url":"https://github.com/optuna/optuna","owner":"optuna","repo":"optuna","owner_avatar_url":"https://avatars.githubusercontent.com/u/57251745?v=4","primary_language":"Python","stars":14603,"forks":1361,"topics":["distributed","hyperparameter-optimization","machine-learning","parallel","python"],"archived":false,"github_pushed_at":"2026-08-03T06:38:06+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/optuna-optuna","markdown_url":"https://www.graphcanon.com/tools/optuna-optuna.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/optuna-optuna","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=optuna-optuna","shared_categories":["model-training"]},{"slug":"huggingface-accelerate","name":"accelerate","tagline":"A tool for launching, training, and using PyTorch models with ease on various devices, configurations, including mixed precision support.","github_url":"https://github.com/huggingface/accelerate","owner":"huggingface","repo":"accelerate","owner_avatar_url":"https://avatars.githubusercontent.com/u/25720743?v=4","primary_language":"Python","stars":9803,"forks":1425,"topics":[],"archived":false,"github_pushed_at":"2026-07-30T11:20:54+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/huggingface-accelerate","markdown_url":"https://www.graphcanon.com/tools/huggingface-accelerate.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/huggingface-accelerate","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=huggingface-accelerate","shared_categories":["model-training","inference-serving"]},{"slug":"hyperopt-hyperopt","name":"hyperopt","tagline":"Distributed Asynchronous Hyperparameter Optimization in Python","github_url":"https://github.com/hyperopt/hyperopt","owner":"hyperopt","repo":"hyperopt","owner_avatar_url":"https://avatars.githubusercontent.com/u/5280805?v=4","primary_language":"Python","stars":7598,"forks":1075,"topics":["hacktoberfest"],"archived":false,"github_pushed_at":"2026-08-03T21:34:53+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/hyperopt-hyperopt","markdown_url":"https://www.graphcanon.com/tools/hyperopt-hyperopt.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/hyperopt-hyperopt","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=hyperopt-hyperopt","shared_categories":["model-training"]},{"slug":"tensorflow-serving","name":"serving","tagline":"A flexible, high-performance serving system for machine learning models","github_url":"https://github.com/tensorflow/serving","owner":"tensorflow","repo":"serving","owner_avatar_url":"https://avatars.githubusercontent.com/u/15658638?v=4","primary_language":"C++","stars":6359,"forks":2204,"topics":["cpp","deep-learning","deep-neural-networks","machine-learning","ml","neural-network","python","serving","tensorflow"],"archived":false,"github_pushed_at":"2026-07-30T07:02:43+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/tensorflow-serving","markdown_url":"https://www.graphcanon.com/tools/tensorflow-serving.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/tensorflow-serving","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=tensorflow-serving","shared_categories":["inference-serving"]},{"slug":"zou-group-textgrad","name":"textgrad","tagline":"Automatic 'Differentiation' via Text using Large Language Models to Backpropagate Textual Gradients","github_url":"https://github.com/zou-group/textgrad","owner":"zou-group","repo":"textgrad","owner_avatar_url":"https://avatars.githubusercontent.com/u/30398052?v=4","primary_language":"Python","stars":3700,"forks":294,"topics":["ai-optimization","compound-systems","large-language-models","prompt-optimization","textual-gradients"],"archived":false,"github_pushed_at":"2025-07-25T14:30:39+00:00","maintenance_label":"Dormant","url":"https://www.graphcanon.com/tools/zou-group-textgrad","markdown_url":"https://www.graphcanon.com/tools/zou-group-textgrad.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/zou-group-textgrad","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=zou-group-textgrad","shared_categories":["model-training"]},{"slug":"keras-team-keras-tuner","name":"keras-tuner","tagline":"A Hyperparameter Tuning Library for Keras","github_url":"https://github.com/keras-team/keras-tuner","owner":"keras-team","repo":"keras-tuner","owner_avatar_url":"https://avatars.githubusercontent.com/u/34455048?v=4","primary_language":"Python","stars":2923,"forks":404,"topics":["automl","deep-learning","hyperparameter-optimization","keras","machine-learning","tensorflow"],"archived":false,"github_pushed_at":"2025-12-01T22:23:25+00:00","maintenance_label":"Slowing","url":"https://www.graphcanon.com/tools/keras-team-keras-tuner","markdown_url":"https://www.graphcanon.com/tools/keras-team-keras-tuner.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/keras-team-keras-tuner","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=keras-team-keras-tuner","shared_categories":["model-training"]},{"slug":"merrymercy-awesome-tensor-compilers","name":"awesome-tensor-compilers","tagline":"A collection of compiler projects and papers for tensor computation and deep learning.","github_url":"https://github.com/merrymercy/awesome-tensor-compilers","owner":"merrymercy","repo":"awesome-tensor-compilers","owner_avatar_url":"https://avatars.githubusercontent.com/u/15100009?v=4","primary_language":null,"stars":2770,"forks":327,"topics":["code-generation","compiler","deep-learning","high-performance-computing","machine-learning","programming-language","tensor"],"archived":false,"github_pushed_at":"2024-10-19T14:35:00+00:00","maintenance_label":"Dormant","url":"https://www.graphcanon.com/tools/merrymercy-awesome-tensor-compilers","markdown_url":"https://www.graphcanon.com/tools/merrymercy-awesome-tensor-compilers.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/merrymercy-awesome-tensor-compilers","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=merrymercy-awesome-tensor-compilers","shared_categories":["model-training","inference-serving"]},{"slug":"microsoft-pai","name":"pai","tagline":"Resource scheduling and cluster management for AI","github_url":"https://github.com/microsoft/pai","owner":"microsoft","repo":"pai","owner_avatar_url":"https://avatars.githubusercontent.com/u/6154722?v=4","primary_language":"JavaScript","stars":2686,"forks":549,"topics":["ai","artificial-intelligence","chainer","cloud","cluster-management","cluster-manager","gpu","gpu-cluster","gpu-computing","gpu-scheduler","jupyter","kubernetes","machine-learning","model-training","on-premise","pytorch","resource-management","scheduling","tensorflow"],"archived":true,"github_pushed_at":"2024-06-06T07:56:07+00:00","maintenance_label":"Archived","url":"https://www.graphcanon.com/tools/microsoft-pai","markdown_url":"https://www.graphcanon.com/tools/microsoft-pai.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/microsoft-pai","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=microsoft-pai","shared_categories":["model-training","inference-serving"]},{"slug":"automl-auto-pytorch","name":"Auto-PyTorch","tagline":"Automatic architecture search and hyperparameter optimization for PyTorch","github_url":"https://github.com/automl/Auto-PyTorch","owner":"automl","repo":"Auto-PyTorch","owner_avatar_url":"https://avatars.githubusercontent.com/u/6469053?v=4","primary_language":"Python","stars":2541,"forks":303,"topics":["automl","deep-learning","pytorch","tabular-data"],"archived":false,"github_pushed_at":"2024-04-09T06:17:08+00:00","maintenance_label":"Dormant","url":"https://www.graphcanon.com/tools/automl-auto-pytorch","markdown_url":"https://www.graphcanon.com/tools/automl-auto-pytorch.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/automl-auto-pytorch","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=automl-auto-pytorch","shared_categories":["model-training"]}]}}