Deep Learning Frameworks

Core deep learning training and inference frameworks — PyTorch, TensorFlow, JAX, and similar stacks.

213 projects

See methodology for ranking rules; order uses public GitHub metrics within this scenario.

81–100 of 213

Rank Project Stars Forks
81 LoRA

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

13.8K 924
82 torchmetrics

Machine learning metrics for distributed, scalable PyTorch applications.

2.5K 518
83 maxtext

A simple, performant, and scalable Jax LLM!

2.4K 593
84 jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

3.2K 324
85 singa

a distributed deep learning platform

3.6K 1.3K
86 optax

Optax is a gradient processing and optimization library for JAX.

2.3K 361
87 aimet

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

2.7K 460
88 equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

3.0K 213
89 tensorflow-onnx

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX

2.6K 472
90 rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

2.9K 605
91 catalyst

Accelerated deep learning R&D

3.4K 397
92 RAGEN

Agent RL framework for LLM agents: multi-turn reinforcement learning with StarPO and reasoning-collapse diagnostics

2.8K 228
93 torch-points3d

Pytorch framework for doing deep learning on point clouds.

2.7K 404
94 xTuring

Build, personalize and control your own LLMs. From data pre-processing to fine-tuning, xTuring provides an easy way to personalize open-source LLMs. Join our discord community: https://discord.gg/TgHXuSJEk6

2.7K 211
95 pipcook

Machine learning platform for Web developers

2.6K 210
96 tiny-cuda-nn

Lightning fast C++/CUDA neural network framework

4.5K 577
97 spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

2.1K 319
98 tnt

A lightweight library for PyTorch training tools and utilities

1.7K 310
99 xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

2.8K 570
100 snntorch

Deep and online learning with spiking neural networks in Python

2.0K 302