# Importing large python npy data sets and doing post processing

**URL:** <https://forum.elixirforum.com/t/importing-large-python-npy-data-sets-and-doing-post-processing/67503>\
**Category:** Questions\
**Tags:** learning-elixir, nx\
**Created:** [November 15, 2024, 6:36pm UTC](https://forum.elixirforum.com/t/importing-large-python-npy-data-sets-and-doing-post-processing/67503 "2024-11-15T18:36:27Z")\
**Posts on this page:** 1\
**Showing post:** 2

<div class="post-metadata">

**Author:** ![billylanchantin](https://forum.elixirforum.com/user_avatar/forum.elixirforum.com/billylanchantin/32/28428_2.png) [@billylanchantin](https://forum.elixirforum.com/u/billylanchantin)\
**Post date:** [November 16, 2024, 3:26pm UTC](https://forum.elixirforum.com/t/importing-large-python-npy-data-sets-and-doing-post-processing/67503/2 "2024-11-16T15:26:40Z")

</div>

Hi @Geegee, welcome to the forum 🙂

In _principle_, yes. `Nx` is designed to work on the GPU and in fact has a least squares function built in:

- [`Nx.LinAlg.least_squares/3`](https://hexdocs.pm/nx/Nx.LinAlg.html#least_squares/3)

The trouble will be getting the data out of the `.npy` format. A quick search turned up this library:

- [Npy — npy v0.1.1](https://hexdocs.pm/npy/0.1.1/Npy.html)

That appears to be exactly what you want. I have not tried that library though so I can’t vouch for it.

---

_[View the full topic](https://forum.elixirforum.com/t/importing-large-python-npy-data-sets-and-doing-post-processing/67503)._