Listed in editorial order, grouped by use case. Click a column to re-sort the whole list.
Press / to search. Tap a tag to filter. Click any row for details.
Search and filter
Results
| Row number | Tags | |||||
|---|---|---|---|---|---|---|
Frameworks 5 projects |
||||||
| 1 | pytorch Frameworks | 60,815,390 | 103,956 | Frameworks Deep Learning AI & ML | → | |
|
Tensors and Dynamic neural networks in Python with strong GPU acceleration.
|
||||||
| 2 | pytorch-lightning Frameworks | 10,156,328 | 31,394 | Frameworks Deep Learning AI & ML | → | |
|
Deep learning framework to train, deploy, and ship AI products Lightning fast.
|
||||||
| 3 | jax Frameworks | 9,311,538 | 36,398 | Frameworks Deep Learning AI & ML | → | |
|
A library for high-performance numerical computing with automatic differentiation and JIT compilation.
|
||||||
| 4 | keras Frameworks | 11,700,449 | 64,357 | Frameworks Deep Learning AI & ML | → | |
|
A high-level deep learning library with support for JAX, TensorFlow, and PyTorch backends.
|
||||||
| 5 | tensorflow Frameworks | 12,531,598 | 200,581 | Frameworks Deep Learning AI & ML | → | |
|
An end-to-end machine learning platform from Google.
|
||||||
Reinforcement Learning 2 projects |
||||||
| 6 | gymnasium Reinforcement Learning | 3,770,179 | 12,643 | Reinforcement Learning Deep Learning AI & ML | → | |
|
A standard API for reinforcement learning environments with popular reference environments (gym successor).
|
||||||
| 7 | stable-baselines3 Reinforcement Learning | 709,977 | 13,880 | Reinforcement Learning Deep Learning AI & ML | → | |
|
PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.
|
||||||
No projects match your search or filter.
Try a broader term, or .
Deep Learning guide
PyTorch runs each line of code as it's reached, so a debugger and stack traces read the way you'd expect. Its quickstart shows the pieces of a project: a Dataset holds your samples and labels, and a DataLoader wraps it in an iterable. A model subclasses nn.Module, defines its layers in __init__, and passes data through them in forward. To keep a trained model, save its state_dict rather than pickling the whole model, which ties the file to your exact classes and directory layout.
PyTorch Lightning organizes PyTorch code so you write the model logic and it handles the engineering. A LightningModule is still an nn.Module, with your training step and configure_optimizers added, and the Trainer runs the loop around it. Moving to more GPUs, to TPUs, or to 16-bit precision is a Trainer argument, with no change to your model code.
JAX pairs a NumPy-like API with composable transformations: jax.grad takes gradients, jax.jit compiles, and jax.vmap vectorizes. They're designed for functionally pure functions, so pass every input as an argument and return every result. The core holds little that's specific to deep learning. For neural networks, add the JAX AI Stack: Flax for models, Optax for optimizers, and Orbax for checkpoints.
Gymnasium, a fork of OpenAI's Gym, is a standard API for reinforcement learning: one interface between learning algorithms and environments, plus reference environments that follow it. Its basic loop comes down to make(), Env.reset(), Env.step(), and Env.render(), and wrappers change an environment without touching its code. For your own problem, subclass gymnasium.Env with observation and action spaces, and register it so gymnasium.make() builds it like a built-in one.
Stable Baselines3 is a set of reliable implementations of RL algorithms in PyTorch, and it assumes you already know some RL. It follows a scikit-learn-like syntax: create a model such as A2C("MlpPolicy", env), call learn(), then predict(). It trains on any environment that follows the Gymnasium interface. Its RL tips pick the algorithm by action space, since DQN handles only discrete actions and SAC only continuous ones.
For more frameworks, papers, and courses, see awesome-deep-learning.
Contribute
Know a project that belongs here?
Tell us what it does and why it stands out.