Skip to content

Jax

jax

Framework for high-performance numerical computing

Runs on tpu

Used for building NLP models

Offers automatic differentiation and GPU/TPU acceleration


*References

On this page