axlearn

Repository facts

Stars

2,374

Forks

409

Open issues and pull requests

6

Primary language

Python

License

Apache License 2.0 (Apache-2.0)

Owner

apple · Organization

Created

Feb 25, 2023

Last push

Jul 8, 2026

Topics

• deep-learning
• jax

Rank #3,187 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Related projects

Projects that share this one’s GitHub topics.

• 81

  Rank 81: keras-team/keras

  Deep Learning for humans

  64,375 starsPythonApache-2.0Last push Oct 16, 2026

• 708

  Rank 708: wandb/wandb

  The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

  11,301 starsPythonMITLast push Oct 17, 2026

• 849

  Rank 849: arogozhnikov/einops

  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

  9,623 starsPythonMITLast push Aug 26, 2026

• 1,078

  Rank 1,078: HIPS/autograd

  Efficiently computes derivatives of NumPy code.

  7,543 starsPythonMITLast push Oct 12, 2026

• 1,791

  Rank 1,791: lululxvi/deepxde

  A library for scientific machine learning and physics-informed learning

  4,471 starsPythonLGPL-2.1Last push Aug 18, 2026

• 2,061

  Rank 2,061: google-research/scenic

  Scenic: A Jax Library for Computer Vision Research and Beyond

  3,844 starsPythonApache-2.0Last push Sep 28, 2026