pytorch-lightning

Lightning-AI/pytorch-lightningFrameworks

Deep learning framework to train, deploy, and ship AI products Lightning fast.

  • python
  • deep-learning
  • artificial-intelligence
  • ai
  • pytorch
  • data-science
  • machine-learning

More in Deep Learning

tensorflowC++ core

An end-to-end machine learning platform from Google.

201.2kupdated today2.21.0

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration.

105kupdated today2.14.1

keras

A high-level deep learning library with support for JAX, TensorFlow, and PyTorch backends.

64.4kupdated 1d ago3.12.4

jax

A library for high-performance numerical computing with autodiff and JIT compilation.

36.5kupdated todayjax-v0.12.0.1

stable-baselines3

PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.

13.9kupdated 11d ago2.9.0

gymnasium

A standard API for reinforcement learning environments with popular reference environments (gym successor).

12.7kupdated today1.4.0
All 7 in Deep Learning
Esc

Type to search every entry. ↑↓ to move, Enter to open.