// readme

DeepJ

DeepJ is a Java library for building, training, and running neural networks. It provides N-dimensional tensors, automatic differentiation, composable layers, and CPU execution with optional Apple Metal acceleration.

Build feed-forward networks, convolutional networks, recurrent networks, autoencoders, and Transformers on the same tensor and autograd foundation.

Website Β· Documentation Β· API reference

Supported networks

Model typeImplementations
Feed-forward networksFNN, with configurable hidden and output activations
Convolutional networksConv1D, Conv2D, max/average pooling, and global average pooling
Recurrent networksRNN, GRU, LSTM, individual cells, explicit state, and per-sample sequence lengths
AutoencodersComposable Autoencoder with separate encoder and decoder access
Transformer encodersBidirectional attention, padding masks, and masked mean pooling
Decoder TransformersDeepJ Origin, Orbit, Prism, Nova, Loop, Qwen3, and Gemma 2 for causal…