google/flax

Flax is a neural network library for JAX that is designed for flexibility.

GitHub repository with 7,225 stars and 814 forks.

Language: Jupyter Notebook

Topics: jax

Open provider repository

24h trend summary

Trending score 0.32, activity score 0.05, stars gained +1, forks gained +1.

Latest metric snapshot

2026-06-05: 7,225 stars and 814 forks.

Similar repositories

  1. 1. google/flax

    Flax is a neural network library for JAX that is designed for flexibility.

    GitHub repository with 7,225 stars and 814 forks.

    Trending score: 0.32; stars gained: +1; forks gained: +1.

    Language: Jupyter Notebook

    Topics: jax

Trending in Jupyter Notebook

  1. 1. GoogleCloudPlatform/generative-ai

    Sample code and notebooks for Generative AI on Google Cloud, with Gemini Enterprise Agent Platform

    GitHub repository with 16,983 stars and 4,249 forks.

    Trending score: 1.87; stars gained: +8; forks gained: +6.

    Language: Jupyter Notebook

    Topics: agents, gcp, gemini, gemini-api, gen-ai, generative-ai

  2. 2. DataTalksClub/llm-zoomcamp

    LLM Zoomcamp - a free online course about real-life applications of LLMs. In 10 weeks you will learn how to build an AI system that answers questions about your knowledge base.

    GitHub repository with 5,624 stars and 1,018 forks.

    Trending score: 1.87; stars gained: +93; forks gained: +13.

    Language: Jupyter Notebook

  3. 3. Biohub/esm

    GitHub repository with 2,691 stars and 332 forks.

    Trending score: 1.82; stars gained: +48; forks gained: +12.

    Language: Jupyter Notebook

  4. 4. nerdai/llm-agents-from-scratch

    Build LLM agents and multi-agent systems from scratch, with MCP, Skills, and A2A

    GitHub repository with 131 stars and 45 forks.

    Trending score: 1.70; stars gained: +32; forks gained: +14.

    Language: Jupyter Notebook

  5. 5. DataTalksClub/data-engineering-zoomcamp

    Data Engineering Zoomcamp is a free 9-week course on building production-ready data pipelines. The next cohort starts in January 2026. Join the course here 👇🏼

    GitHub repository with 41,902 stars and 8,303 forks.

    Trending score: 1.49; stars gained: +23; forks gained: +3.

    Language: Jupyter Notebook

    Topics: course, data-engineering, dbt, docker, free, kafka

  6. 6. facebookresearch/VLM3

    Official implementation of paper "VLM³: Vision Language Models Are Native 3D Learners".

    GitHub repository with 218 stars and 9 forks.

    Trending score: 1.46; stars gained: +33; forks gained: +0.

    Language: Jupyter Notebook

    Topics: 3d-foundation-model, camera-pose-estimation, depth-estimation, image-matching, large-language-models, object-level-3d

Trending topic: jax

  1. 1. jax-ml/jax

    Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

    GitHub repository with 35,760 stars and 3,610 forks.

    Trending score: 1.23; stars gained: +18; forks gained: +1.

    Language: Python

    Topics: jax

  2. 2. keras-team/keras

    Deep Learning for humans

    GitHub repository with 64,088 stars and 19,747 forks.

    Trending score: 0.65; stars gained: +3; forks gained: -1.

    Language: Python

    Topics: deep-learning, tensorflow, neural-networks, machine-learning, data-science, python

  3. 3. PennyLaneAI/pennylane

    PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

    GitHub repository with 3,247 stars and 810 forks.

    Trending score: 0.60; stars gained: +3; forks gained: +14.

    Language: Python

    Topics: autograd, automatic-differentiation, cirq, deep-learning, differentiable-computing, jax

  4. 4. PASSIONLab/OpenEquivariance

    OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

    GitHub repository with 149 stars and 9 forks.

    Trending score: 0.45; stars gained: +1; forks gained: +0.

    Language: Python

    Topics: cuda, geometric-deep-learning, graph-neural-networks, sparse-tensors, equivariance, hip

  5. 5. google/flax

    Flax is a neural network library for JAX that is designed for flexibility.

    GitHub repository with 7,225 stars and 814 forks.

    Trending score: 0.32; stars gained: +1; forks gained: +1.

    Language: Jupyter Notebook

    Topics: jax

  6. 6. team-ocean/veros

    The versatile ocean simulator, in pure Python, powered by JAX.

    GitHub repository with 395 stars and 59 forks.

    Trending score: 0.19; stars gained: +0; forks gained: +0.

    Language: Python

    Topics: oceanography, python, parallel, multi-core, geophysics, jax