Skip to content

0.1.0

Choose a tag to compare

View all tags
@petrpravda petrpravda released this 18 Oct 13:49
· 6 commits to main since this release
v0.1.0
0056f1e

Initial release of JAX4J: Java 25 tensors, automatic differentiation and neural networks.

Included

  • CPU tensor execution, fused JIT, batching, structured transforms, autodiff, neural-network layers and Adam optimizers.
  • Vulkan and wgpu-native tensor engines, with shared GPU kernels and device-resident execution.
  • Consolidated CUDA and ROCm runtime/replay modules, optional SIMD replay, and an MNIST MLP example.
  • Seven module JARs and SHA-256 checksums. GitHub also supplies source archives.

Validation and requirements

All modules passed mvn -Pcuda,rocm,examples,simd -Djax4j.vulkan.tests=true -Djax4j.wgpu.tests=true clean verify: 126 tests passed, 1 existing test skipped, with no failures or errors. Vulkan and wgpu hardware suites were enabled. CUDA/ROCm compilation and packaging passed; their native execution remains unverified.

Java 25 is required. GPU native libraries are installed separately; they are not included in the JARs. Native execution uses --enable-native-access=ALL-UNNAMED; SIMD additionally requires --add-modules jdk.incubator.vector. Tensor execution is currently F32; the remaining roadmap and limitations are documented in implementation status.

Build from source

git clone --branch v0.1.0 --depth 1 https://github.com/nablatensor-dev/jax4j.git
cd jax4j
mvn -Pcuda,rocm,examples,simd -DskipTests install

Maven group: com.nablatensor; version: 0.1.0. Applications normally use jax4j-core with the desired engine; jax4j is the parent POM. This release is distributed through GitHub and is not published to Maven Central.

See native setup, README, and the release guide.

Maven Central

Version 0.1.0 is now available on Maven Central under com.nablatensor.
Use jax4j-core:0.1.0 plus the desired engine artifact. The jax4j artifact is the parent POM.

Central's main JARs match these GitHub release binaries byte for byte. Sources, Javadocs and signatures are also published. The existing tag is unchanged. Finance APIs are maintained in NablaTensor Quant.

Assets 10
Loading