brainstate

discover · compare · collections · requirements · contribute · feed · watching · stats · help with your tokens

arogozhnikov/einops

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

Python★ 9.6kMITGitHub ↗

History and source changes · Reports and owner responses · Compare

Similar

  1. unifyai/ivy

    Convert Machine Learning Code Between Frameworks

    Python★ 14.2kGitHub ↗

    Why this match

    topics jax, numpy, pytorch

  2. keras-team/keras

    Deep Learning for humans

    Python★ 64.4kApache-2.0GitHub ↗

    Why this match

    topics deep-learning, jax, pytorch; license differs: MIT → Apache-2.0

  3. aymericdamien/TopDeepLearning

    A list of popular github projects related to deep learning

    Python★ 6.4kMITGitHub ↗

    Why this match

    topics deep-learning, pytorch, tensorflow

  4. google/trax

    Trax — Deep Learning with Clear Code and Speed

    Python★ 8.3kApache-2.0archivedGitHub ↗

    Why this match

    topics deep-learning, jax, numpy; license differs: MIT → Apache-2.0

  5. pytorch/pytorch

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python★ 105.1kGitHub ↗

    Why this match

    topics deep-learning, numpy, tensor

  6. jax-ml/jax

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

    Python★ 36.6kApache-2.0GitHub ↗

    Why this match

    topics jax; license differs: MIT → Apache-2.0

  7. ml-explore/mlx

    MLX: An array framework for Apple silicon

    C++★ 29.1kMITGitHub ↗

    Why this match

    topics mlx; language differs: Python → C++

  8. ml-explore/mlx-examples

    Examples in the MLX framework

    Python★ 9.1kMITGitHub ↗

    Why this match

    topics mlx