pytorch

pytorch/pytorchFrameworks

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

  • neural-network
  • autograd
  • gpu
  • numpy
  • deep-learning
  • tensor
  • python
  • machine-learning

More in Deep Learning

tensorflowC++ core

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

201.2kupdated today2.21.0

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

pytorch-lightning

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

31.4kupdated 12d ago2.6.6

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.