{"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":36338,"forks":3810,"topics":["jax"],"archived":false,"github_pushed_at":"2026-09-26T05:38:44+00:00","maintenance_label":"Very active","stars_delta_30d":253,"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":64343,"forks":19806,"topics":["data-science","deep-learning","jax","machine-learning","neural-networks","python","pytorch","tensorflow"],"archived":false,"github_pushed_at":"2026-09-26T04:22:33+00:00","maintenance_label":"Very active","stars_delta_30d":126,"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":200493,"forks":77664,"topics":["deep-learning","deep-neural-networks","distributed","machine-learning","ml","neural-network","python","tensorflow"],"archived":false,"github_pushed_at":"2026-09-27T05:55:11+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":103384,"forks":30659,"topics":["autograd","deep-learning","gpu","machine-learning","neural-network","numpy","python","tensor"],"archived":false,"github_pushed_at":"2026-09-27T05:57:49+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":31363,"forks":3801,"topics":["ai","artificial-intelligence","data-science","deep-learning","machine-learning","python","pytorch"],"archived":false,"github_pushed_at":"2026-09-21T11:44:30+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":14870,"forks":1406,"topics":["distributed","hyperparameter-optimization","machine-learning","parallel","python"],"archived":false,"github_pushed_at":"2026-09-30T07:26:24+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":9885,"forks":1513,"topics":[],"archived":false,"github_pushed_at":"2026-09-25T16:32:10+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":7595,"forks":1074,"topics":["hacktoberfest"],"archived":false,"github_pushed_at":"2026-09-28T22:18:00+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":"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":2921,"forks":406,"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":2788,"forks":330,"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":2694,"forks":554,"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":false,"github_pushed_at":"2026-09-18T03:17:30+00:00","maintenance_label":"Active","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":2540,"forks":298,"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"]},{"slug":"google-parfait-tensorflow-federated","name":"tensorflow-federated","tagline":"An open-source framework for machine learning and other computations on decentralized data","github_url":"https://github.com/google-parfait/tensorflow-federated","owner":"google-parfait","repo":"tensorflow-federated","owner_avatar_url":"https://avatars.githubusercontent.com/u/164364956?v=4","primary_language":"Python","stars":2454,"forks":605,"topics":[],"archived":false,"github_pushed_at":"2026-10-01T22:43:59+00:00","maintenance_label":"Active","url":"https://www.graphcanon.com/tools/google-parfait-tensorflow-federated","markdown_url":"https://www.graphcanon.com/tools/google-parfait-tensorflow-federated.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/google-parfait-tensorflow-federated","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=google-parfait-tensorflow-federated","shared_categories":["model-training"]}]}}