jax4j-core
64 public types across 14 packages, generated from the engine repo’s own Javadoc comments.
com.nablatensor.jax4j.annotation
- Internalannotation
com.nablatensor.jax4j.backend.cpu
- CpuBackendclass
com.nablatensor.jax4j.engine
- AadCheckpointPlanclass
- AadEngineinterface
- AadEnginesclass
- AadExecutableinterface
- AadOpenum
- AadOptionsrecord
- AadRecorderclass
- AadResultclass
- AadTapeclass
- AbstractAadExecutableclass
- CudaAadCodegenclass
- JitOptimizationsclass
- Nablaclass
- SDoubleclass
com.nablatensor.jax4j.engine.cpu
- AadCpuReplayclass
- CpuAadEngineclass
- Philoxclass
com.nablatensor.jax4j.engine.jit
- JitAadEngineclass
- JitKernelinterface
- JitKernelF32interface
- JitPhiloxclass
com.nablatensor.jax4j.nn
- Adamclass
- Denseclass
- FunctionalAdamclass
- MaskedPoolingclass
com.nablatensor.jax4j.tensor
- Backendenum
- BackendRegistryclass
- ControlFlowclass
- ConvSpecrecord
- CustomDerivativeclass
- Devicerecord
- DeviceProgramclass
- DeviceTypeenum
- DTypeenum
- Jitclass
- Linalgclass
- Openum
- PrngKeyrecord
- QuantTypeenum
- Shaperecord
- Tensorclass
- Tensorsclass
- Transformsclass
- Vmapclass
com.nablatensor.jax4j.tensor.bench
- GpuComparisonclass
com.nablatensor.jax4j.tensor.expr
- Exprinterface
com.nablatensor.jax4j.tensor.linalg
- DenseLinalgclass
com.nablatensor.jax4j.tensor.spi
- ComputeBackendinterface
- DeviceBufferinterface
- GpuBufferclass
- GpuRuntimeinterface
- GpuTensorBackendclass
- QuantizedOpsinterface
- TracingBackendinterface
- TracingBufferinterface