Python Software Engineer

Job Summary

We are seeking a talented and versatile Python Software Engineer to join our team. In this role, you will be responsible for developing and maintaining machine learning models. In addition, we will also be building User interfaces to visualize the input data using various plots For e.g. FFT. The ideal candidate will have strong expertise in both building and optimizing deep learning models using PyTorch Lightning, as well as a passion for creating intuitive, user-friendly graphical interfaces with Tkinter. This position requires a strong understanding of data processing, model deployment, and software engineering best practices.

Key Responsibilities

• Application Development: Design, develop, and maintain cross-platform desktop/cli applications using Python and the Tkinter library.
• Machine Learning Integration: Integrate and deploy machine learning models within the desktop/cli application for tasks such as data analysis, prediction, or classification.
• Model Development: Research, train, and fine-tune deep learning models using PyTorch Lightning. Ability to also do pre and post quantization trainings for various bitness. Knowledge on integer – fixed and floating point formats is an added advantage
• Data Processing: Utilize NumPy and Pandas to efficiently handle, preprocess, and analyze large datasets for model training and application functionality.
• Visualization: Various charts native as well as exported charts via matplot lib.
• Testing and Debugging Utility development: Write and execute driver programs to run the trainer, inference and pre/post process various forms of data.
• Collaboration: Work closely with managers, cross functional teams and other engineers to gather requirements and deliver high-quality solutions.
• Documentation: Maintain thorough documentation of code, APIs, and ML models.

Required Skills and Qualifications

• Bachelor's or Master's degree in Computer Science, Software Engineering, or a related technical field.
• Proven experience as a Python Software Engineer and must be hands-on. Ability to use pip or uv package managers is a must.
• Demonstrated experience with PyTorch Lightning for building, training deep learning models, loading/saving models, serialize/deserialize model weights, bias and other hyper-parameters.
• Expertise with Tkinter for developing graphical user interfaces. Good to know websockets/SSE to interface with remote components.
• Extensive experience with the Pandas and NumPy libraries for data manipulation and numerical operations.
• Solid understanding of the entire machine learning lifecycle, from data preparation to model deployment.
• Proficiency with version control tools, such as Git.
• Strong problem-solving, analytical, and debugging skills with VSCode as the primary editor
• Excellent communication and collaboration abilities.
• Knowledge on signal analysis is an added advantage
• Ability to integrate other python libraries on smart devices, controllers is an added advantage – For e.g. Pynq libraries